def _clip_gradients(self, max_norm):
    nn.utils.clip_grad_norm_(master_params(self.optimizer), max_norm)
def _clip_gradients(self, max_norm):
    nn.utils.clip_grad_norm_(master_params(self.optimizer), max_norm)