steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

CCS intergration dataloader

Browse Source
This commit is contained in:
Steffen 2021-03-19 18:05:17 +01:00
parent 43cf0ad00d
commit 675312537f
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/config.py
View File

@ -70,7 +70,11 @@ def parse_comandline_args_add_defaults(filepath, overrides=None):
log_save_interval=10000, # TODO: Better Value / Setting
auto_lr_find=not args['debug'],
weights_summary='top',
check_val_every_n_epoch=1 if args['debug'] else args.get('check_val_every_n_epoch', 1)
check_val_every_n_epoch=1 if args['debug'] else args.get('check_val_every_n_epoch', 1),
limit_train_batches = 2.0,
limit_val_batches = 2.0,
limit_test_batches = 2.0,
limit_predict_batches = 2.0,
)
if overrides is not None and isinstance(overrides, (Mapping, Dict)):