fix typo

bomtorazek · Mar 31, 2021 · d27dd1c · d27dd1c
1 parent b42de11
commit d27dd1c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/two_stream_bert/utils.py b/two_stream_bert/utils.py
@@ -1,5 +1,6 @@
 import shutil
 import os
+import torch
 
 class AverageMeter(object):
     """Computes and stores the average and current value"""