This code looks good. But it seems that we have to train two models if we don’t freeze the pretrained layers. This ways requires more memory more time. If I have more memory, why do I train deeper network, like resnet 101.