Model not training

[ptrblck]

If you want to replace the last linear layer in model_conv.classifier, you would have to reassign the new nn.Sequential module back to .classifier instead of .classifier1.