=-



 ptrblck:

Could you remove the sigmoid as the last non-linearity, as nn.CrossEntropyLoss expects raw logits


I have tried this. It didn’t work. I remember you’ve recommended this before, so I’ve already tried it.



 ptrblck:

Also note that you are passing vgg_comp_model.parameters() to the optimizer, which is undefined in your code.


Apologies for this. Kindly change the name to cifar_model as defined.