• eellison's avatar
    Make maskrcnn scriptable (#1407) · d88d8961
    eellison authored
    * almost working...
    
    * respond to comments
    
    * add empty tensor op, handle different output types in generalized rcnn
    
    * clean ups
    
    * address comments
    
    * more changes
    
    * it's working!
    
    * torchscript bugs
    
    * add script/ eager test
    
    * eval script model
    
    * fix flake
    
    * division import
    
    * py2 compat
    
    * update test, fix arange bug
    
    * import division statement
    
    * fix linter
    
    * fixes
    
    * changes needed for JIT master
    
    * cleanups
    
    * remove imagelist_to
    
    * requested changes
    
    * Make FPN backwards-compatible and torchscript compatible
    
    We remove support for feature channels=0, but support for it was already a bit limited
    
    * Fix ONNX regression
    d88d8961
conf.py 7.97 KB