Description
2023-04-02 22:47:42,200 INFO Start training cfgs/once_models/VFL3D(default)
epochs: 0%| | 0/100 [00:02<?, ?it/s]
Traceback (most recent call last): | 0/1241 [00:00<?, ?it/s]
File "train.py", line 229, in
main()
File "train.py", line 198, in main
use_amp=args.use_amp
File "/home/libing/newstorage1/2/OpenPCDetnow/tools/train_utils/train_utils.py", line 187, in train_model
use_amp=use_amp
File "/home/libing/newstorage1/2/OpenPCDetnow/tools/train_utils/train_utils.py", line 56, in train_one_epoch
loss, tb_dict, disp_dict = model_func(model, batch)
File "../pcdet/models/init.py", line 42, in model_func
ret_dict, tb_dict, disp_dict = model(batch_dict)
File "/home/libing/anaconda3/envs/HVPR/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "../pcdet/models/detectors/rdiou_net.py", line 14, in forward
loss, tb_dict, disp_dict = self.get_training_loss()
File "../pcdet/models/detectors/rdiou_net.py", line 27, in get_training_loss
loss_rpn, tb_dict = self.dense_head.get_loss()
File "../pcdet/models/dense_heads/anchor_head_rdiou_3cat.py", line 335, in get_loss
self.pse_box_cls_labels = self.get_clsreg_targets()
File "../pcdet/models/dense_heads/anchor_head_rdiou_3cat.py", line 183, in get_clsreg_targets
_, rdiou = self.get_rdiou(box_preds, box_reg_targets)
File "../pcdet/models/dense_heads/anchor_head_rdiou_3cat.py", line 87, in get_rdiou
t1 = torch.sin(bboxes1[:,:,6]) * torch.cos(bboxes2[:,:,6])
RuntimeError: The size of tensor a (212064) must match the size of tensor b (3393024) at non-singleton dimension 1