Tutorial-about-3D-convolutional-network icon indicating copy to clipboard operation
Tutorial-about-3D-convolutional-network copied to clipboard
verified publisher icon OValery16

model training

Open fatiherbag opened this issue 6 years ago • 1 comments

I have 8 labels in my dataset. I get mismatch error when i try training.

size mismatch for module.fc7.weight: copying a param of torch.Size([8, 256]) from checkpoint, where the shape is torch.Size([27, 256]) in current model.

fatiherbag avatar Jan 09 '20 11:01 fatiherbag

do not include this layer in the module you restore via model.load_state_dict(...)

OValery16 avatar Feb 27 '20 10:02 OValery16