Pārlūkot izejas kodu

Update general.py (#823)

Fixes #822

`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
5.0
Willie Maddox GitHub pirms 4 gadiem
vecāks
revīzija
455f7b8f76
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 3 papildinājumiem un 2 dzēšanām
  1. +3
    -2
      utils/general.py

+ 3
- 2
utils/general.py Parādīt failu

@@ -23,7 +23,8 @@ from scipy.cluster.vq import kmeans
from scipy.signal import butter, filtfilt
from tqdm import tqdm

from utils.torch_utils import init_seeds, is_parallel
from utils.torch_utils import init_seeds as init_torch_seeds
from utils.torch_utils import is_parallel

# Set printoptions
torch.set_printoptions(linewidth=320, precision=5, profile='long')
@@ -55,7 +56,7 @@ def set_logging(rank=-1):
def init_seeds(seed=0):
random.seed(seed)
np.random.seed(seed)
init_seeds(seed=seed)
init_torch_seeds(seed=seed)


def get_latest_run(search_dir='./runs'):

Notiek ielāde…
Atcelt
Saglabāt