- 25 Nov, 2019 1 commit
-
-
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
-
- 15 Nov, 2019 1 commit
-
-
eellison authored
* remove changes that induced BC * Re-enable tests that have been disabled * Remove outdated comment * Remove outdated comment
-
- 20 Sep, 2019 1 commit
-
-
eellison authored
* script_fcn_resnet * Make old models load * DeepLabV3 also got torchscript-ready
-
- 20 May, 2019 1 commit
-
-
Francisco Massa authored
* Add more documentation for the ops * Add documentation for Faster R-CNN * Add documentation for Mask R-CNN and Keypoint R-CNN * Improve doc for RPN * Add basic doc for GeneralizedRCNNTransform * Lint fixes
-
- 10 May, 2019 1 commit
-
-
Francisco Massa authored
* Initial version of the segmentation examples WIP * Cleanups * [WIP] * Tag where runs are being executed * Minor additions * Update model with new resnet API * [WIP] Using torchvision datasets * Improving datasets Leverage more and more torchvision datasets * Reorganizing datasets * PEP8 * No more SegmentationModel Also remove outplanes from ResNet, and add a function for querying intermediate outputs. I won't keep it in the end, because it's very hacky and don't work with tracing * Minor cleanups * Moving transforms to its own file * Move models to torchvision * Bugfixes * Multiply LR by 10 for classifier * Remove classifier x 10 * Add tests for segmentation models * Update with latest utils from classification * Lint and missing import
-