still fixing checkpointing

alex_trainL.py

@@ -102,8 +102,8 @@ def train_model(args, model_id, dataset_id):
 
         # Naive 80/20 (change this I guess)
         #print(len(seq_datasets[0]))
-        train_data = ConcatDataset(seq_datasets[:14]) # 14
-        val_data = ConcatDataset(seq_datasets[14:])
+        train_data = ConcatDataset(seq_datasets[:1]) # 14
+        val_data = ConcatDataset(seq_datasets[:1])
 
         print(f"train: {seq_names[:14]}")
         print(f"val: {seq_names[14:]}")