diff --git a/train.py b/train.py index 7e91a77a..361dc2c1 100644 --- a/train.py +++ b/train.py @@ -14,7 +14,6 @@ import torch.nn as nn from torch import optim from torch.autograd import Variable from torch.utils.data import DataLoader -from torch.optim.lr_scheduler import ReduceLROnPlateau from tensorboardX import SummaryWriter from utils.generic_utils import (Progbar, remove_experiment_folder,