def freeze_all_except_classifiers(self): self.freeze_weights_extractor() freeze_weights(self.fconv_block)
def freeze_weights_extractor(self): freeze_weights(self.conv_block1) freeze_weights(self.conv_block2) freeze_weights(self.conv_block3) freeze_weights(self.conv_block4) freeze_weights(self.conv_block5)