Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
With the mps device set, you can:
train larger networks or batch sizes locally
reduce data retrieval latency because the GPU's unified memory architecture allows direct access to the full memory store
reduce costs because you don't need to train on cloud-based GPUs or add additional local GPUs
Get started by making sure you have PyTorch installed.