-
Notifications
You must be signed in to change notification settings - Fork 13
Description
Great job! But I have a question: When I run train_vcoco.sh, there is a problem that:
Traceback (most recent call last):
File "main.py", line 619, in
main(args)
File "main.py", line 515, in main
args.gradient_accumulation_steps, args.enable_amp, args.no_training, args)
File "/usr1/home/s124mdg44_05/HOICLIP-main/HOICLIP-main/engine.py", line 71, in train_one_epoch
outputs = model(samples, clip_input=clip_img, targets=targets)
File "/usr1/home/s124mdg44_05/miniforge3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/usr1/home/s124mdg44_05/miniforge3/envs/py37/lib/python3.7/site-packages/torch/nn/parallel/distributed.py", line 619, in forward
output = self.module(*inputs[0], **kwargs[0])
File "/usr1/home/s124mdg44_05/miniforge3/envs/py37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/usr1/home/s124mdg44_05/HOICLIP-main/HOICLIP-main/models/models_hoiclip/hoiclip.py", line 206, in forward
outputs_verb_class = logit_scale * self.verb_projection(verb_hs) @ self.verb2hoi_proj
RuntimeError: mat1 dim 1 must match mat2 dim 0
Can you help me fix it? Thank you!