Skip to content

GitLab

Commit bc29c9e2 authored by dcuai's avatar dcuai
Browse files

Update config.py ——add epochs

parent 60aeae99
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
config.py
View file @ bc29c9e2
...@@ -291,6 +291,10 @@ def update_config(config, args): ...@@ -291,6 +291,10 @@ def update_config(config, args):
# merge from specific arguments # merge from specific arguments
if _check_args('batch_size'): if _check_args('batch_size'):
config.DATA.BATCH_SIZE = args.batch_size config.DATA.BATCH_SIZE = args.batch_size
if _check_args('train_epochs'):
config.TRAIN.EPOCHS = args.train_epochs
if _check_args('warmup_epochs'):
config.TRAIN.WARMUP_EPOCHS = args.warmup_epochs
if _check_args('data_path'): if _check_args('data_path'):
config.DATA.DATA_PATH = args.data_path config.DATA.DATA_PATH = args.data_path
if _check_args('zip'): if _check_args('zip'):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment