Ejemplo n.º 1
0
 def freeze_all_except_classifiers(self):
     self.freeze_weights_extractor()
     freeze_weights(self.fconv_block)
Ejemplo n.º 2
0
 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)