From 37067fad05fc1f27e92dfe7189bfd0151881fac1 Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Tue, 24 Mar 2020 19:13:06 +0800
Subject: [PATCH] Fix constant typo
---
src/core/factories.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/core/factories.py b/src/core/factories.py
index 5a35789..23198c7 100644
--- a/src/core/factories.py
+++ b/src/core/factories.py
@@ -173,7 +173,7 @@ def _get_basic_configs(ds_name, C):
)
elif ds_name.startswith('AC'):
return dict(
- root = constants.IMDB_AirChange
+ root = constants.IMDB_AIRCHANGE
)
elif ds_name.startswith('Lebedev'):
return dict(
@@ -247,7 +247,7 @@ def single_val_ds_factory(ds_name, C):
dataset_obj,
batch_size=1,
shuffle=False,
- num_workers=C.num_workers,
+ num_workers=1,
pin_memory=False, drop_last=False
)
--
GitLab