Fix import on initalization

2018-11-03 15:38:30 -04:00
parent ec66841afa
commit cd09cd5b40
2 changed files with 2 additions and 0 deletions
--- a/run_classifier_pytorch.py
+++ b/run_classifier_pytorch.py
@@ -23,6 +23,7 @@ import os
 import logging
 import argparse

+import random
 import numpy as np
 from tqdm import tqdm, trange
 import torch