From 675312537ff627b2e1b0ef005800fdfdb7fc89b9 Mon Sep 17 00:00:00 2001 From: Steffen Date: Fri, 19 Mar 2021 18:05:17 +0100 Subject: [PATCH] CCS intergration dataloader --- utils/config.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/utils/config.py b/utils/config.py index 5111dfe..87b3b17 100644 --- a/utils/config.py +++ b/utils/config.py @@ -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)):