Asking  question out of context @ptrblck , how can I change the require_grad of each layer after getting the self.features ?
I can make requires_grad to False by iterating self.model.parameters() but what in case I extract feature part of VGG16 (to modify it) and later want to change requires_grad. Because I don’t see any option such as self.model.parameters() after extracting self.features.