File size: 520 Bytes
f6d6286 c735a0e f6d6286 c735a0e f6d6286 c735a0e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
# from model_core.training import train_memgpt
#from model_core.dataloader import DataLoader
from model_core.training import train_memgpt
from model_core.dataloader import DataLoader
if __name__ == "__main__":
config_path = "configs/config.json"
print("Training starter")
train_memgpt(config_path=config_path,dataloader_class=DataLoader)
#to run the code using ddp: torchrun --nproc-per-node=1 train.py |