Update evaluate_source_network_on_evaluation.py

UlysseCoteAllard · May 14, 2019 · 985279b · 985279b
1 parent 9184785
commit 985279b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/PyTorchImplementation/RawEnhancedConvNet/evaluate_source_network_on_evaluation.py b/PyTorchImplementation/RawEnhancedConvNet/evaluate_source_network_on_evaluation.py
@@ -238,17 +238,17 @@ def train_model(cnn, criterion, optimizer, scheduler, dataloaders, num_epochs=50
 
     # Comment between here
 
-    examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="training0")
+    examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="training0")
     datasets = [examples, labels]
 
     np.save("formatted_datasets/saved_evaluation_dataset_training.npy", datasets)
 
-    examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="Test0")
+    examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="Test0")
     datasets = [examples, labels]
 
     np.save("formatted_datasets/saved_evaluation_dataset_test0.npy", datasets)
 
-    examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="Test1")
+    examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="Test1")
     datasets = [examples, labels]
 
     np.save("formatted_datasets/saved_evaluation_dataset_test1.npy", datasets)