• Yanghan Wang's avatar
    Allow specifying extra lightning trainer params via `_DEFAULTS_` in yaml · 6940fa9c
    Yanghan Wang authored
    Summary:
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/461
    
    There're needs for extending trainer parameters that are not in (or conflict with) the base d2go config, this diff adds a way to inject those configs without touching the base d2go config.
    - In `get_trainer_params`, it simply checks the `LIGHTNING_TRAINER` and use whatever configs under it.
    - Adds `GeneralizedRCNNTaskNoDefaultConfig`, which allows specify default config via yaml file for `GeneralizedRCNNTask`. (also make some changes for prerequisite)
    - (next diff) User can add their own config updater by registering it in `CONFIG_UPDATER_REGISTRY`.
    
    Differential Revision: D42928992
    
    fbshipit-source-id: f2a1d8a3f2bec9908bb1af03928611d963b92c0e
    6940fa9c
lightning_train_net.py 6.2 KB