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

Small bugfixes

Browse Source
This commit is contained in:
Steffen
2021-03-18 12:12:43 +01:00
parent fecf4923c2
commit 2c9cb2e94a
2 changed files with 108 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -10,12 +10,16 @@ from ml_lib.utils.config import parse_comandline_args_add_defaults
from ml_lib.utils.loggers import Logger
import variables as v
from ml_lib.utils.tools import fix_all_random_seeds
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
def run_lightning_loop(h_params, data_class, model_class, additional_callbacks=None):
def run_lightning_loop(h_params, data_class, model_class, seed=69, additional_callbacks=None):
fix_all_random_seeds(seed)
with Logger.from_argparse_args(h_params) as logger:
# Callbacks
# =============================================================================
@ -79,13 +83,13 @@ def run_lightning_loop(h_params, data_class, model_class, additional_callbacks=N
if __name__ == '__main__':
# Parse comandline args, read config and get model
cmd_args, found_data_class, found_model_class = parse_comandline_args_add_defaults('_parameters.ini')
cmd_args, found_data_class, found_model_class, found_seed = parse_comandline_args_add_defaults('_parameters.ini')
# To NameSpace
hparams = Namespace(**cmd_args)
# Start
# -----------------
run_lightning_loop(hparams, found_data_class, found_model_class)
run_lightning_loop(hparams, found_data_class, found_model_class, found_seed)
print('done')
pass