Hi, first thanks for your work. When I run the 2d train code, it reminds me the following errors. I don't know how to solve this. Could you help me find the problems?
Epoch: [0][0/696] Time 5.216s (5.216s) Speed 6.1 samples/s Data 2.286s (2.286s) Loss 0.36521 (0.36521) Accuracy 0.026 (0.026) Memory 1429799424.0
Epoch: [0][100/696] Time 0.424s (0.543s) Speed 75.4 samples/s Data 0.000s (0.041s) Loss 0.34272 (0.39108) Accuracy 0.000 (0.002) Memory 1429799424.0
Epoch: [0][200/696] Time 0.453s (0.518s) Speed 70.7 samples/s Data 0.000s (0.028s) Loss 0.37073 (0.37585) Accuracy 0.015 (0.002) Memory 1428750848.0
Epoch: [0][300/696] Time 0.495s (0.506s) Speed 64.7 samples/s Data 0.000s (0.023s) Loss 0.34146 (0.36804) Accuracy 0.133 (0.016) Memory 1429799424.0
Epoch: [0][400/696] Time 0.503s (0.500s) Speed 63.6 samples/s Data 0.000s (0.020s) Loss 0.26815 (0.35295) Accuracy 0.424 (0.076) Memory 1428750848.0
Epoch: [0][500/696] Time 0.488s (0.495s) Speed 65.5 samples/s Data 0.000s (0.018s) Loss 0.25259 (0.33488) Accuracy 0.460 (0.149) Memory 1429799424.0
Epoch: [0][600/696] Time 0.423s (0.492s) Speed 75.7 samples/s Data 0.000s (0.017s) Loss 0.23425 (0.31868) Accuracy 0.551 (0.212) Memory 1428750848.0
Traceback (most recent call last):
File "run/pose2d/train.py", line 189, in
main()
File "run/pose2d/train.py", line 164, in main
criterion, final_output_dir, writer_dict)
File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/core/function.py", line 233, in validate
name_value, perf_indicator = dataset.evaluate(all_preds)
File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/dataset/multiview_h36m.py", line 145, in evaluate
distance = np.sqrt(np.sum((gt - pred)**2, axis=2))
ValueError: operands could not be broadcast together with shapes (0,) (0,17,2)
Hi, first thanks for your work. When I run the 2d train code, it reminds me the following errors. I don't know how to solve this. Could you help me find the problems?
Epoch: [0][0/696] Time 5.216s (5.216s) Speed 6.1 samples/s Data 2.286s (2.286s) Loss 0.36521 (0.36521) Accuracy 0.026 (0.026) Memory 1429799424.0 Epoch: [0][100/696] Time 0.424s (0.543s) Speed 75.4 samples/s Data 0.000s (0.041s) Loss 0.34272 (0.39108) Accuracy 0.000 (0.002) Memory 1429799424.0 Epoch: [0][200/696] Time 0.453s (0.518s) Speed 70.7 samples/s Data 0.000s (0.028s) Loss 0.37073 (0.37585) Accuracy 0.015 (0.002) Memory 1428750848.0 Epoch: [0][300/696] Time 0.495s (0.506s) Speed 64.7 samples/s Data 0.000s (0.023s) Loss 0.34146 (0.36804) Accuracy 0.133 (0.016) Memory 1429799424.0 Epoch: [0][400/696] Time 0.503s (0.500s) Speed 63.6 samples/s Data 0.000s (0.020s) Loss 0.26815 (0.35295) Accuracy 0.424 (0.076) Memory 1428750848.0 Epoch: [0][500/696] Time 0.488s (0.495s) Speed 65.5 samples/s Data 0.000s (0.018s) Loss 0.25259 (0.33488) Accuracy 0.460 (0.149) Memory 1429799424.0 Epoch: [0][600/696] Time 0.423s (0.492s) Speed 75.7 samples/s Data 0.000s (0.017s) Loss 0.23425 (0.31868) Accuracy 0.551 (0.212) Memory 1428750848.0 Traceback (most recent call last): File "run/pose2d/train.py", line 189, in
main()
File "run/pose2d/train.py", line 164, in main
criterion, final_output_dir, writer_dict)
File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/core/function.py", line 233, in validate
name_value, perf_indicator = dataset.evaluate(all_preds)
File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/dataset/multiview_h36m.py", line 145, in evaluate
distance = np.sqrt(np.sum((gt - pred)**2, axis=2))
ValueError: operands could not be broadcast together with shapes (0,) (0,17,2)