Added few naming changes

model_core/dataloader.py

@@ -3,7 +3,7 @@ import numpy as np
 import torch
 
 #Data loader
-class DataLoaderLite:
+class DataLoader_1:
     def __init__(self, B, T, process_rank, num_processes, split, master_process):
         self.B = B
         self.T = T

scripts/train.py

@@ -3,9 +3,9 @@ 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 DataLoaderLite
+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=DataLoaderLite)
+    train_memgpt(config_path=config_path,dataloader_class=DataLoader_1)