qq456cvb / Point-Transformers

Point Transformers
MIT License
609 stars 102 forks source link

gradient disappears #45

Open chaibosong opened 2 months ago

chaibosong commented 2 months ago

The batchsize is set to 32. On the 1x4090 machine, the accuracy reaches 80% for 22 epochs, and then the accuracy drops sharply to 10%.

2024-05-15 14:29:15,914][main][INFO] - Train Instance Accuracy: 0.758549 [2024-05-15 14:29:32,168][main][INFO] - Test Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:29:32,168][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:29:32,168][main][INFO] - Save model... [2024-05-15 14:29:32,168][main][INFO] - Saving at best_model.pth [2024-05-15 14:29:32,203][main][INFO] - Epoch 22 (22/200): [2024-05-15 14:30:32,547][main][INFO] - Train Instance Accuracy: 0.760876 [2024-05-15 14:30:48,515][main][INFO] - Test Instance Accuracy: 0.722756, Class Accuracy: 0.663108 [2024-05-15 14:30:48,515][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:30:48,515][main][INFO] - Epoch 23 (23/200): [2024-05-15 14:31:48,781][main][INFO] - Train Instance Accuracy: 0.749724 [2024-05-15 14:32:04,739][main][INFO] - Test Instance Accuracy: 0.141426, Class Accuracy: 0.103739 [2024-05-15 14:32:04,739][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:32:04,740][main][INFO] - Epoch 24 (24/200): [2024-05-15 14:33:04,951][main][INFO] - Train Instance Accuracy: 0.735230 [2024-05-15 14:33:20,807][main][INFO] - Test Instance Accuracy: 0.326522, Class Accuracy: 0.314507 [2024-05-15 14:33:20,807][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:33:20,807][main][INFO] - Epoch 25 (25/200): [2024-05-15 14:34:20,965][main][INFO] - Train Instance Accuracy: 0.728553 [2024-05-15 14:34:37,029][main][INFO] - Test Instance Accuracy: 0.294471, Class Accuracy: 0.286097 [2024-05-15 14:34:37,029][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:34:37,029][main][INFO] - Epoch 26 (26/200): [2024-05-15 14:35:37,281][main][INFO] - Train Instance Accuracy: 0.717668 [2024-05-15 14:35:53,427][main][INFO] - Test Instance Accuracy: 0.190705, Class Accuracy: 0.181524 [2024-05-15 14:35:53,427][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:35:53,427][main][INFO] - Epoch 27 (27/200): [2024-05-15 14:36:53,845][main][INFO] - Train Instance Accuracy: 0.711399 [2024-05-15 14:37:09,817][main][INFO] - Test Instance Accuracy: 0.379808, Class Accuracy: 0.343976 [2024-05-15 14:37:09,817][main][INFO] - Best Instance Accuracy: 0.860176, Class Accuracy: 0.819621 [2024-05-15 14:37:09,817][main][INFO] - Epoch 28 (28/200):