Fix docker execution

.dockerignore

@@ -0,0 +1,2 @@
+data
+work_dirs

predict.py

@@ -6,8 +6,8 @@ from pathlib import Path
 
 
 # custom script arguments
-CONFIG_PATH = 'work_dirs/swin_base_b32x4-fp16_fungi+val_res_384_cb_epochs_6/swin_base_b32x4-fp16_fungi+val-test_res_384_cb_epochs_6.py'
-CHECKPOINT_PATH = "work_dirs/swin_base_b32x4-fp16_fungi+val_res_384_cb_epochs_6/epoch_6.pth"
+CONFIG_PATH = 'models/swin_base_b32x4-fp16_fungi+val_res_384_cb_epochs_6.py'
+CHECKPOINT_PATH = "models/swin_base_b32x4-fp16_fungi+val_res_384_cb_epochs_6_20230524-5197a7e6.pth"
 SCORE_THRESHOLD = 0.2
 
 

tools/test_generate_result_pre-consensus.py

@@ -129,7 +129,7 @@ def setup_multi_processes(cfg):
 
     # setup OMP threads
     # This code is referred from https://github.com/pytorch/pytorch/blob/master/torch/distributed/run.py  # noqa
-    if 'OMP_NUM_THREADS' not in os.environ and cfg.data.workers_per_gpu > 1:
+    if 'OMP_NUM_THREADS' not in os.environ and cfg.test_dataloader.num_workers > 1:
         omp_num_threads = 1
         warnings.warn(
             f'Setting OMP_NUM_THREADS environment variable for each process '
@@ -139,7 +139,7 @@ def setup_multi_processes(cfg):
         os.environ['OMP_NUM_THREADS'] = str(omp_num_threads)
 
     # setup MKL threads
-    if 'MKL_NUM_THREADS' not in os.environ and cfg.data.workers_per_gpu > 1:
+    if 'MKL_NUM_THREADS' not in os.environ and cfg.test_dataloader.num_workers > 1:
         mkl_num_threads = 1
         warnings.warn(
             f'Setting MKL_NUM_THREADS environment variable for each process '