def set_mode(self, mode): self.mode = mode for layer in self.middle_layers: layer.set_mode(mode) self.cost_layer.set_mode(mode)