From d4e3cf352023e62fbda3e16c2e302979df40fee0 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sun, 4 Nov 2018 10:54:16 +0100 Subject: [PATCH] add numpy import --- run_squad.py | 1 + 1 file changed, 1 insertion(+) diff --git a/run_squad.py b/run_squad.py index fa7575de25..46c689496c 100644 --- a/run_squad.py +++ b/run_squad.py @@ -28,6 +28,7 @@ import os from tqdm import tqdm, trange import random +import numpy as np import torch from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler from torch.utils.data.distributed import DistributedSampler