Searched refs:trainable_state (Results 1 – 2 of 2) sorted by relevance
2109 trainable_state = {self: self.trainable}2111 trainable_state.update(l._get_trainable_state())2112 return trainable_state2114 def _set_trainable_state(self, trainable_state): argument2116 if self in trainable_state:2117 self.trainable = trainable_state[self]2120 if l in trainable_state:2121 l._set_trainable_state(trainable_state)
2758 trainable_state = weakref.WeakKeyDictionary()2760 trainable_state[layer] = layer.trainable2761 return trainable_state2763 def _set_trainable_state(self, trainable_state): argument2766 if layer in trainable_state:2767 layer.trainable = trainable_state[layer]