Browse Source
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
4 years ago
1 changed files with 3 additions and 2 deletions
+ 3
- 2
utils/general.py
View File
| from scipy.signal import butter, filtfilt | from scipy.signal import butter, filtfilt | ||||
| from tqdm import tqdm | 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 | # Set printoptions | ||||
| torch.set_printoptions(linewidth=320, precision=5, profile='long') | torch.set_printoptions(linewidth=320, precision=5, profile='long') | ||||
| def init_seeds(seed=0): | def init_seeds(seed=0): | ||||
| random.seed(seed) | random.seed(seed) | ||||
| np.random.seed(seed) | np.random.seed(seed) | ||||
| init_seeds(seed=seed) | |||||
| init_torch_seeds(seed=seed) | |||||
| def get_latest_run(search_dir='./runs'): | def get_latest_run(search_dir='./runs'): |
Loading…