def __freeze__(self): for layer in layers.get_all_layers(self.net): for param in layer.params: layer.params[param].discard('trainable')