Update trainer.py

This commit is contained in:
William Falcon
2019-07-26 08:23:37 -04:00
committed by GitHub
parent 7d97e3e6e4
commit ba25161dcc
-1
View File
@@ -10,7 +10,6 @@ import re
import torch
from torch.utils.data.distributed import DistributedSampler
from torch.optim.lr_scheduler import MultiStepLR
import torch.multiprocessing as mp
import torch.distributed as dist
import numpy as np