ptrblck:

[None, None, None, :]


I was trying to remove dimensions in the loading, but I wasn’t quite sure what had the wrong dimensions and I entirely forgot about np.squeeze(), so I was trying to use flatten, which didn’t help. layer.state_dict()['bias'].copy_(np.squeeze(torch.from_numpy(_weights_dict[name]['bias']))) works and loads correctly.