add numpy import

This commit is contained in:
thomwolf 2018-11-04 10:54:16 +01:00
parent cf366417d5
commit d4e3cf3520

View File

@ -28,6 +28,7 @@ import os
from tqdm import tqdm, trange from tqdm import tqdm, trange
import random import random
import numpy as np
import torch import torch
from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
from torch.utils.data.distributed import DistributedSampler from torch.utils.data.distributed import DistributedSampler