option to run io as argparse

deepskies · Mar 28, 2024 · 0412dfc · 0412dfc
1 parent 1a8e079
commit 0412dfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scripts/io.py b/src/scripts/io.py
@@ -343,4 +343,4 @@ def train_val_split(model_inputs,
     trainSteps = len(trainDataLoader.dataset) // BATCH_SIZE
     valSteps = len(valDataLoader.dataset) // BATCH_SIZE
     '''
-    return trainDataLoader, x_val, y_val
+    #return trainDataLoader, x_val, y_val