classifier, export, torch seed updates
This commit is contained in:
1 parent
c5d2331897
commit
883924d9dc
3 files changed
+22
-18
No files matched your search
+2
-3
@@ -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'):
|
||||
|
||||
Reference in new issue
Block a user