diff --git a/main.py b/main.py index 07caa23043011ec203c9a082222b94a11d81d1d7..f21ba8b0697409ad2189813c71262b385a471518 100644 --- a/main.py +++ b/main.py @@ -95,8 +95,8 @@ if __name__ == "__main__": valid_ratio, overwrite_index = True, max_num_samples=max_num_samples, - train_transform= eval(dataset_transfom), - valid_transform=eval(dataset_transfom) + train_transform= eval(dataset_transform), + valid_transform=eval(dataset_transform) ) if use_cuda : @@ -135,4 +135,4 @@ if __name__ == "__main__": utils.write_summary(logdir, network, optimizer, val_loss) - create_submission.create_submission(network, eval(dataset_transfom), device, rootDir, logdir) + create_submission.create_submission(network, eval(dataset_transform), device, rootDir, logdir)