• Mashiro's avatar
    [Enhancement] Enhance get_torchvision_models (#1867) · a80df687
    Mashiro authored
    * enhance get_torchvision_models
    
    * simplify logic
    
    * Dump ckpt in torchvision lower than 0.13.0 to a json file
    
    * add json
    
    * refactor load urls logic
    
    * fix unit test
    
    * change url key to lower letters
    
    * check torchvision version rather than check torch version in unittest
    
    * Fix CI and refine test logic of torchvision version
    
    * add comment
    
    * support compare pre-release version
    
    * support loaad modeel like torchvision
    
    * refine comment.
    
    * fix unit test and comment
    
    * fxi unit test bug
    
    * support get model by lower weights
    a80df687
test_load_model_zoo.py 5.87 KB