Thanks ptrblck. I’m working on a dataset class right now. Loss does not seem to decrease at all. I will implement transforms.Normalize in the dataset class.
This is for cell detection on a noisy background: Just to get the thing rolling I made fake training images like this

The background is always the same, so I’d be surprised if normalizing helped