Set num_workers to 0

fedeizzo · Jan 4, 2022 · b006b4c · b006b4c
1 parent 9444d5d
commit b006b4c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -64,7 +64,7 @@ outputs = 7
 # save preprocessed dataset according to data folder structure.
 # This reduces the dataset loading time on each run
 save_processed_dataset = true
-num_workers = 4
+num_workers = 0
 batch_size = 128
 
 [loss]
@@ -159,7 +159,7 @@ feature_dimension = 2048
 dropout_rate = 0.5
 
 [dataloader]
-num_workers = 4
+num_workers = 0
 batch_size = 64
 # value used for torchvision jitter transform
 color_jitter = 0.5

diff --git a/camera-pose-estimation/model/mapnet_train.ini.sample b/camera-pose-estimation/model/mapnet_train.ini.sample
@@ -10,7 +10,7 @@ feature_dimension = 2048
 dropout_rate = 0.5
 
 [dataloader]
-num_workers = 4
+num_workers = 0
 batch_size = 16
 step = 3
 skip = 10

diff --git a/camera-pose-estimation/model/menet_train.ini.sample b/camera-pose-estimation/model/menet_train.ini.sample
@@ -9,7 +9,7 @@ name = "menet"
 outputs = 7
 
 [dataloader]
-num_workers=4
+num_workers = 0
 batch_size = 64
 
 [loss]

diff --git a/camera-pose-estimation/model/posenet_train.ini.sample b/camera-pose-estimation/model/posenet_train.ini.sample
@@ -9,7 +9,7 @@ name = "posenet"
 outputs = 7
 
 [dataloader]
-num_workers=4
+num_workers = 0
 batch_size = 64
 
 [loss]