def _set_model_train_mode(self):
    'Set train mode for model'
    phase = self.phases[self.phase_idx]
    self.base_model.train(phase['train'])
    self.base_loss.train(phase['train'])
    if ((self.broadcast_buffers_mode == BroadcastBuffersMode.BEFORE_EVAL) and (not self.train)):
        self._broadcast_buffers()
Set train mode for model
def _set_model_train_mode(self):
    
    phase = self.phases[self.phase_idx]
    self.base_model.train(phase['train'])
    self.base_loss.train(phase['train'])
    if ((self.broadcast_buffers_mode == BroadcastBuffersMode.BEFORE_EVAL) and (not self.train)):
        self._broadcast_buffers()