导入预训练模型last.pt训练,报错信息:
Traceback (most recent call last):
File "train.py", line 547, in
train() # train normally
File "train.py", line 392, in train
optimizer.step()
File "/usr/local/lib/python3.5/dist-packages/torch/optim/sgd.py", line 100, in step
buf.mul(momentum).add(1 - dampening, d_p)
The size of tensor a (512) must match the size of tensor b (256) at non-singleton dimension 0
导入预训练模型last.pt训练,报错信息: Traceback (most recent call last): File "train.py", line 547, in
train() # train normally
File "train.py", line 392, in train
optimizer.step()
File "/usr/local/lib/python3.5/dist-packages/torch/optim/sgd.py", line 100, in step
buf.mul(momentum).add(1 - dampening, d_p)
The size of tensor a (512) must match the size of tensor b (256) at non-singleton dimension 0