thank you, that help me, but when i have the error
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 11, 11], but got 5-dimensional input of size [20, 16, 3, 224, 224] instead
if i try call unsqueeze(0) i get [ 1,20,16,224,224] how can i reduce the dimension to 4 ??