From 10d72390c029b3f139639621fb9a3a264560e05b Mon Sep 17 00:00:00 2001 From: Lysandre Date: Fri, 22 May 2020 10:49:45 -0400 Subject: [PATCH] Revert #4446 Since it introduces a new dependency --- src/transformers/trainer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 4362832644..47d7c91f67 100644 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -11,7 +11,6 @@ from typing import Callable, Dict, List, Optional, Tuple import numpy as np import torch -from packaging import version from torch import nn from torch.utils.data.dataloader import DataLoader from torch.utils.data.dataset import Dataset @@ -495,11 +494,8 @@ class Trainer: ): logs: Dict[str, float] = {} logs["loss"] = (tr_loss - logging_loss) / self.args.logging_steps - # backward compatibility for pytorch schedulers logs["learning_rate"] = ( scheduler.get_last_lr()[0] - if version.parse(torch.__version__) >= version.parse("1.4") - else scheduler.get_lr()[0] ) logging_loss = tr_loss