From 2eb297878edbb388ef5a663292346fd94c60e653 Mon Sep 17 00:00:00 2001 From: Yandi <yandirzm@gmail.com> Date: Sat, 4 Feb 2023 13:38:14 +0100 Subject: [PATCH] fixing type (missing r) --- main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 07caa23..f21ba8b 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) -- GitLab