Megvii-BaseDetection / BEVDepth

Official code for BEVDepth.
MIT License
688 stars 96 forks source link

Allow specifying training epochs when calling `run_cli` #139

Closed Francis777 closed 1 year ago

Francis777 commented 1 year ago

The number of training epochs seems to be hardcoded to 24 and isn't overwritten anywhere(maybe I'm missing something?), this PR adds a new input argument extra_trainer_config_args to run_cli, I think it's better than directly adding epochs as input because it's more flexible and can be used to specify other trainer args in the future.

The default behavior is unchanged.

yinchimaoliang commented 1 year ago

Thanks for your contribution!