W1031 13:20:55.689000 1 torch/distributed/run.py:851] W1031 13:20:55.689000 1 torch/distributed/run.py:851] ***************************************** W1031 13:20:55.689000 1 torch/distributed/run.py:851] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. W1031 13:20:55.689000 1 torch/distributed/run.py:851] ***************************************** Using Device: cpu, Backend: gloo Using Device: cpu, Backend: gloo Distributed Training for WORLD_SIZE: 4, RANK: 0, LOCAL_RANK: 0 Distributed Training for WORLD_SIZE: 4, RANK: 1, LOCAL_RANK: 1 Using dataset from shared PVC at /mnt/sharedUsing dataset from shared PVC at /mnt/shared Train Epoch: 1 [0/60000 (0%)] Loss: 2.302340 Train Epoch: 1 [1000/60000 (7%)] Loss: 1.915588 Train Epoch: 1 [2000/60000 (13%)] Loss: 1.951827 Train Epoch: 1 [3000/60000 (20%)] Loss: 1.558424 Train Epoch: 1 [4000/60000 (27%)] Loss: 0.855369 Train Epoch: 1 [5000/60000 (33%)] Loss: 0.865366 Train Epoch: 1 [6000/60000 (40%)] Loss: 0.787873 Train Epoch: 1 [7000/60000 (47%)] Loss: 0.856324 Train Epoch: 1 [8000/60000 (53%)] Loss: 0.484378 Train Epoch: 1 [9000/60000 (60%)] Loss: 0.610479 Train Epoch: 1 [10000/60000 (67%)] Loss: 0.396351 Train Epoch: 1 [11000/60000 (73%)] Loss: 0.461476 Train Epoch: 1 [12000/60000 (80%)] Loss: 0.457438 Train Epoch: 1 [13000/60000 (87%)] Loss: 0.584209 Train Epoch: 1 [14000/60000 (93%)] Loss: 0.412306 Train Epoch: 2 [0/60000 (0%)] Loss: 0.506281 Train Epoch: 2 [1000/60000 (7%)] Loss: 0.424659 Train Epoch: 2 [2000/60000 (13%)] Loss: 0.450444 Train Epoch: 2 [3000/60000 (20%)] Loss: 0.311605 Train Epoch: 2 [4000/60000 (27%)] Loss: 0.384921 Train Epoch: 2 [5000/60000 (33%)] Loss: 0.323043 Train Epoch: 2 [6000/60000 (40%)] Loss: 0.529112 Train Epoch: 2 [7000/60000 (47%)] Loss: 0.529589 Train Epoch: 2 [8000/60000 (53%)] Loss: 0.334968 Train Epoch: 2 [9000/60000 (60%)] Loss: 0.382584 Train Epoch: 2 [10000/60000 (67%)] Loss: 0.262031 Train Epoch: 2 [11000/60000 (73%)] Loss: 0.347102 Train Epoch: 2 [12000/60000 (80%)] Loss: 0.325355 Train Epoch: 2 [13000/60000 (87%)] Loss: 0.318753 Train Epoch: 2 [14000/60000 (93%)] Loss: 0.273543 Training is finished