Just in case, these are my parameters:
|2021-08-09 20:40:58,801 NN-Training INFO: |seed: 543|
|2021-08-09 20:40:58,802 NN-Training INFO: |output_path: test|
|2021-08-09 20:40:58,803 NN-Training INFO: |momentum: 0.9|
|2021-08-09 20:40:58,804 NN-Training INFO: |weight_decay: 0.0001|
|2021-08-09 20:40:58,805 NN-Training INFO: |num_workers: 12|
|2021-08-09 20:40:58,806 NN-Training INFO: |num_epochs: 5|
|2021-08-09 20:40:58,807 NN-Training INFO: |learning_rate: 0.05|
|2021-08-09 20:40:58,809 NN-Training INFO: |num_warmup_epochs: 1|
|2021-08-09 20:40:58,810 NN-Training INFO: |validate_every: 1|
|2021-08-09 20:40:58,811 NN-Training INFO: |checkpoint_every: 3|
|2021-08-09 20:40:58,813 NN-Training INFO: |backend: gloo|
|2021-08-09 20:40:58,818 NN-Training INFO: |resume_from: None|
|2021-08-09 20:40:58,819 NN-Training INFO: |log_every_iters: 15|
|2021-08-09 20:40:58,820 NN-Training INFO: |nproc_per_node: 4|
|2021-08-09 20:40:58,821 NN-Training INFO: |stop_iteration: None|
|2021-08-09 20:40:58,822 NN-Training INFO: |with_clearml: False|
|2021-08-09 20:40:58,823 NN-Training INFO: |with_amp: False|
|2021-08-09 20:40:58,824 NN-Training INFO: |start_method: fork|