Are you processing c_out somehow afterwards?
Note that different models might use different names for the “last” layer, e.g. model.fc instead of model.classifier.
A batch size of 1 with batch norm layers is usually not a good idea. You could try to replace the batch norm layers by e.g. instance norm, change the momentum argument, remove them, etc.