classifier, export, torch seed updates

This commit is contained in:
Glenn Jocher committed 2020-10-06 14:54:02 +02:00
1 parent c5d2331897
commit 883924d9dc
3 files changed
+22 -18

No files matched your search

+2 -3
View File
@@ -23,8 +23,7 @@ from scipy.signal import butter, filtfilt
from tqdm import tqdm
from utils.google_utils import gsutil_getsize
from utils.torch_utils import init_seeds as init_torch_seeds
from utils.torch_utils import is_parallel
from utils.torch_utils import is_parallel, init_torch_seeds
# Set printoptions
torch.set_printoptions(linewidth=320, precision=5, profile='long')
@@ -56,7 +55,7 @@ def set_logging(rank=-1):
def init_seeds(seed=0):
random.seed(seed)
np.random.seed(seed)
init_torch_seeds(seed=seed)
init_torch_seeds(seed)
def get_latest_run(search_dir='./runs'):