GPT-124m / scripts /train.py
abhinavv3's picture
Added few naming changes
7cec32f
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_1
if __name__ == "__main__":
config_path = "configs/config.json"
print("Training starter")
train_memgpt(config_path=config_path,dataloader_class=DataLoader_1)