From 3080943cde7c10f94bc092cee95ab199faa09660 Mon Sep 17 00:00:00 2001
From: ah02299 <ah02299@surrey.ac.uk>
Date: Fri, 16 Feb 2024 10:59:06 +0000
Subject: [PATCH] still fixing checkpointing
---
alex_trainL.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/alex_trainL.py b/alex_trainL.py
index 084d953..e5f802d 100644
--- a/alex_trainL.py
+++ b/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:]}")
--
GitLab