steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of gitlab.lrz.de:mobile-ifi/inter_challenge_2020

Browse Source
This commit is contained in:
Steffen Illium 2020-05-15 09:35:19 +02:00
commit c1c367ace7
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -20,6 +20,15 @@ warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning) warnings.filterwarnings('ignore', category=UserWarning)
def fix_all_random_seeds(config_obj):
import numpy as np
import torch
import random
np.random.seed(config.main.seed)
torch.manual_seed(config.main.seed)
random.seed(config.main.seed)
def run_lightning_loop(config_obj): def run_lightning_loop(config_obj):
# Logging # Logging
@ -124,4 +133,5 @@ if __name__ == "__main__":
from _paramters import main_arg_parser from _paramters import main_arg_parser
config = MConfig.read_argparser(main_arg_parser) config = MConfig.read_argparser(main_arg_parser)
fix_all_random_seeds(config)
trained_model = run_lightning_loop(config) trained_model = run_lightning_loop(config)