-
Notifications
You must be signed in to change notification settings - Fork 37
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
RuntimeError: mat1 and mat2 shapes cannot be multiplied (6144x1456 and 2720x85) #24
Comments
How is your config file set? |
project_name: tea_swin_t |
same question in here, backbone is EfficientNet |
did you solve this issue |
Solved it! |
use 323232 is OK
|
When I select efficentnet for training I get the following error, only swin-transformer does not report it
Can you help me?
Start Training 1 EpochTraceback (most recent call last):
File "D:/hxy/FGVC-PIM-master/main.py", line 301, in
main(args, tlogger)
File "D:/hxy/FGVC-PIM-master/main.py", line 253, in main
train(args, epoch, model, scaler, amp_context, optimizer, schedule, train_loader)
File "D:/hxy/FGVC-PIM-master/main.py", line 140, in train
outs = model(datas)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\parallel\data_parallel.py", line 168, in forward
outputs = self.parallel_apply(replicas, inputs, kwargs)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\parallel\data_parallel.py", line 178, in parallel_apply
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\parallel\parallel_apply.py", line 86, in parallel_apply
output.reraise()
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch_utils.py", line 457, in reraise
raise exception
RuntimeError: Caught RuntimeError in replica 0 on device 0.
Original Traceback (most recent call last):
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\parallel\parallel_apply.py", line 61, in _worker
output = module(*input, **kwargs)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\hxy\FGVC-PIM-master\models\pim_module\pim_module.py", line 414, in forward
comb_outs = self.combiner(selects)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\hxy\FGVC-PIM-master\models\pim_module\pim_module.py", line 81, in forward
hs = self.param_pool0(hs)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\mj\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\linear.py", line 103, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (6144x1456 and 2720x85)
The text was updated successfully, but these errors were encountered: