• Yanghan Wang's avatar
    update RCNN model test base · 0ab6d3f1
    Yanghan Wang authored
    Summary:
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/75
    
    Refactor the base test case
    - make test_dir valid throughout the test (rather than under local context), so individual test can load back the export model
    - refactor the `custom_setup_test` for easier override.
    - move parameterized into base class to avoid copying naming function
    
    Reviewed By: zhanghang1989
    
    Differential Revision: D28651067
    
    fbshipit-source-id: c59a311564f6114039e20ed3a23e5dd9c84f4ae4
    0ab6d3f1
setup.py 2.91 KB