Skip to content

RuntimeError: mat1 dim 1 must match mat2 dim 0 #24

@Liuyungeng123

Description

@Liuyungeng123

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions