• eellison's avatar
    make shufflenet and resnet scriptable (#1270) · 26c9630b
    eellison authored
    * make shufflenet scriptable
    
    * make resnet18 scriptable
    
    * set downsample to identity instead of __constants__ api
    
    * use __constants__ for downsample instead of identity
    
    * import tensor to fix flake
    
    * use torch.Tensor type annotation instead of import
    26c9630b
test_models.py 6.54 KB