diff --git a/models/rank/fibinet/config.yaml b/models/rank/fibinet/config.yaml index 4f0951682e4e96c2fb7c4c373a56de4e6d6bc951..d9ae592f3321d46aa221648d309b8d2e2944f53a 100644 --- a/models/rank/fibinet/config.yaml +++ b/models/rank/fibinet/config.yaml @@ -102,9 +102,9 @@ phase: - name: phase1 model: "{workspace}/model.py" # user-defined model dataset_name: dataloader_train # select dataset by name - thread_num: 8 + thread_num: 1 - name: phase2 model: "{workspace}/model.py" # user-defined model dataset_name: dataset_infer # select dataset by name - thread_num: 8 + thread_num: 1