Skip to content
Snippets Groups Projects
Commit 050fb5db authored by cocoalex00's avatar cocoalex00
Browse files

Merge branch 'edenn+' of https://gitlab.surrey.ac.uk/ah02299/edenn into edenn+

parents e5a3fe4e aeda4853
No related branches found
No related tags found
No related merge requests found
......@@ -86,8 +86,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(train_data)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment