• Kai Zhang's avatar
    Simplify Lightning task and model creation · 9051f71a
    Kai Zhang authored
    Summary:
    Given that the way to create D2 (https://github.com/facebookresearch/d2go/commit/465cdb842513eb910aa20fcedea1d2edd15dc7b7)go runner and Lightning task are different, get_class was introduced so that in application we could do:
    ```
    if is Lightning:
        task_cls = get_class(classname)
        task = task_cls(cfg)
    else:
        runner = create_runner(classname)
    ```
    It turns out that we could need to do that in many places: workflow, binaries.
    This diff revert `get_class` and return class in `create_runner` if the class is a Lightning module.
    
    Reviewed By: newstzpz
    
    Differential Revision: D26676595
    
    fbshipit-source-id: c3ce2016d09fe073af4c2dd9f98eea4e59ca621b
    9051f71a
test_runner_default_runner.py 13.5 KB