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