import random bug fix (#614)

5 years ago · eb99dff9ef
parent d5b6416c87
commit eb99dff9ef
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -2,9 +2,9 @@ import argparse
 import glob
 import math
 import os
+import random
 import time
 from pathlib import Path
-from random import random

 import numpy as np
 import torch.distributed as dist