Open lisaner000 opened 6 months ago
It seems like there are some mistakes in data loading. Because it's wired, the shape of pred_theta is [0, 85]. You should debug to check what went wrong.
It seems like there are some mistakes in data loading. Because it's wired, the shape of pred_theta is [0, 85]. You should debug to check what went wrong.
It achieves the training successfully. Thank you very much!
Another, I met a bug during the evaluting as the following:
Traceback (most recent call last):
File "evaluate.py", line 94, in
Traceback (most recent call last): File "train.py", line 168, in
main(cfg)
File "train.py", line 159, in main
debug_freq=cfg.DEBUG_FREQ,
File "/root/autodl-tmp/STAF-main/lib/core/trainer.py", line 400, in fit
self.train() # 调用 train 方法进行模型训练
File "/root/autodl-tmp/STAF-main/lib/core/trainer.py", line 227, in train
scores=scores,
File "/root/miniconda3/envs/yzy_staf/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/STAF-main/lib/core/loss.py", line 116, in forward
pred_theta = pred_theta[w_smpl]
IndexError: The shape of the mask [39] at index 0 does not match the shape of the indexed tensor [0, 85] at index 0
Can you give me some advice?