mmdetection
mmdetection copied to clipboard
I want to know how to set Epoch and Batch_size, when I train my custom model
Hi! Thanks for the great sharing! I want to know how to set Epoch and Batch_size, when I train my custom model I search all the .py file in configs related to my config python file, I did not find any word relative to "epoch" or "batch_size"
Thanks a lot!!
https://github.com/open-mmlab/mmdetection/blob/master/docs/en/tutorials/config.md
the main codes are in configs/base, for example:
https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/configs/base/schedules/schedule_1x.py#L11 https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/configs/base/datasets/coco_detection.py#L32
https://github.com/open-mmlab/mmdetection/blob/master/docs/en/tutorials/config.md
the main codes are in configs/base, for example:
https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/configs/base/schedules/schedule_1x.py#L11
https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/configs/base/datasets/coco_detection.py#L32
Thanks