Thanks @ptrblck, that helped  I’m a bit confused on how to use the aux classifier for training this model. Is it essential to use? Do you know anywhere there’s a quick explanation?
I’m trying to decide which layers to train and at what LR for segmenting a facial dataset of 2000 samples, using 11 classes. Any advice/tips on how to approach would be highly appreciated