Minor fixes of imports.

subset_indices.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:972615a5b506b5ee2490f61866c26a4a2f9e2498c0baedb195a2a0d10a62e76f
+size 111016

trainingModel/run_training.py

@@ -3,7 +3,7 @@ from clearml import Task
 from dataPrep.helpers.clearml_data import extract_latest_data_task
 
 import torch
-from models.modelTwo import modelTwo, modelOne
+from models.modelTwo import BetterCNN
 from trainingModel.helpers.Training import train_model
 
 
@@ -37,7 +37,7 @@ training_task.connect(training_config)
 
 
 # -------- Build the ML model --------
-model = modelTwo(noOfClasses=training_config["num_classes"])
+model = BetterCNN(noOfClasses=training_config["num_classes"])
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)