".github/vscode:/vscode.git/clone" did not exist on "a718345a8d60c73a441f6254d6eae456c8a6d787"
  • Krzysztof Chalupka's avatar
    Replace pluggable components to create a proper Configurable hierarchy. · 1b0584f7
    Krzysztof Chalupka authored
    Summary:
    This large diff rewrites a significant portion of Implicitron's config hierarchy. The new hierarchy, and some of the default implementation classes, are as follows:
    ```
    Experiment
        data_source: ImplicitronDataSource
            dataset_map_provider
            data_loader_map_provider
        model_factory: ImplicitronModelFactory
            model: GenericModel
        optimizer_factory: ImplicitronOptimizerFactory
        training_loop: ImplicitronTrainingLoop
            evaluator: ImplicitronEvaluator
    ```
    
    1) Experiment (used to be ExperimentConfig) is now a top-level Configurable and contains as members mainly (mostly new) high-level factory Configurables.
    2) Experiment's job is to run factories, do some accelerate setup and then pass the results to the main training loop.
    3) ImplicitronOptimizerFactory and ImplicitronModelFactory are new high-level factories that create the optimizer, scheduler, model, and stats objects.
    4) TrainingLoop is a new configurable that runs the main training loop and the inner train-validate step.
    5) Evaluator is a new configurable that TrainingLoop uses to run validation/test steps.
    6) GenericModel is not the only model choice anymore. Instead, ImplicitronModelBase (by default instantiated with GenericModel) is a member of Experiment and can be easily replaced by a custom implementation by the user.
    
    All the new Configurables are children of ReplaceableBase, and can be easily replaced with custom implementations.
    
    In addition, I added support for the exponential LR schedule, updated the config files and the test, as well as added a config file that reproduces NERF results and a test to run the repro experiment.
    
    Reviewed By: bottler
    
    Differential Revision: D37723227
    
    fbshipit-source-id: b36bee880d6aa53efdd2abfaae4489d8ab1e8a27
    1b0584f7
model_factory.py 7.16 KB