Skip to content

Commit 985279b

Browse files
Update evaluate_source_network_on_evaluation.py
1 parent 9184785 commit 985279b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

PyTorchImplementation/RawEnhancedConvNet/evaluate_source_network_on_evaluation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,17 +238,17 @@ def train_model(cnn, criterion, optimizer, scheduler, dataloaders, num_epochs=50
238238

239239
# Comment between here
240240

241-
examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="training0")
241+
examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="training0")
242242
datasets = [examples, labels]
243243

244244
np.save("formatted_datasets/saved_evaluation_dataset_training.npy", datasets)
245245

246-
examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="Test0")
246+
examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="Test0")
247247
datasets = [examples, labels]
248248

249249
np.save("formatted_datasets/saved_evaluation_dataset_test0.npy", datasets)
250250

251-
examples, labels = load_evaluation_dataset.read_data('EvaluationDataset', type="Test1")
251+
examples, labels = load_evaluation_dataset.read_data('../../EvaluationDataset', type="Test1")
252252
datasets = [examples, labels]
253253

254254
np.save("formatted_datasets/saved_evaluation_dataset_test1.npy", datasets)

0 commit comments

Comments
 (0)