Ah I see! . I haven’t gotten it working yet. I haven’t figured out why original code is working and not my revised code. There must be something wrong with my revised code. I will play around with a little more and see if I can figure it out.
EDIT:
Even turning almost all layers for training:
for param in model_ft.features.parameters():
    param.requires_grad = False

not resulting in improvements.