Fix import on initalization

This commit is contained in:
VictorSanh 2018-11-03 15:38:30 -04:00
parent ec66841afa
commit cd09cd5b40
2 changed files with 2 additions and 0 deletions

View File

@ -23,6 +23,7 @@ import os
import logging import logging
import argparse import argparse
import random
import numpy as np import numpy as np
from tqdm import tqdm, trange from tqdm import tqdm, trange
import torch import torch

View File

@ -26,6 +26,7 @@ import json
import math import math
import os import os
from tqdm import tqdm, trange from tqdm import tqdm, trange
import random
import torch import torch
from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler