mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 10:13:59 +08:00
4f4f2957ef
* Use build_runner in train api * Support iter in eval_hook * Add runner section * Add test_eval_hook * Pin mmcv version in install docs * Replace max_iters with max_epochs * Set by_epoch=True as default * Remove trailing space * Replace DeprecationWarning with UserWarning * pre-commit * Fix tests
7 lines
250 B
Python
7 lines
250 B
Python
# optimizer
|
|
optimizer = dict(type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0001)
|
|
optimizer_config = dict(grad_clip=None)
|
|
# learning policy
|
|
lr_config = dict(policy='CosineAnnealing', min_lr=0)
|
|
runner = dict(type='EpochBasedRunner', max_epochs=100)
|