- 30 Sep, 2019 1 commit
-
-
Philip Meier authored
-
- 27 Sep, 2019 2 commits
-
-
eellison authored
* make googlnet scriptable * Remove typing import in favor of torch.jit.annotations * add inceptionnet * flake fixes * fix asssert true * add import division for torchscript * fix script compilation * fix flake, py2 division error * fix py2 division error
-
peterjc123 authored
-
- 26 Sep, 2019 1 commit
-
-
Francisco Massa authored
* Fix nightly builds * Fix lint * Disable video_reader tests on Travis * Disable one more test that segfaults on Travis * Fix typo in expression
-
- 24 Sep, 2019 6 commits
-
-
ekosman authored
* Expose num_workers in VideoClips * add documentation for num_workers in VideoClips * add documentation for num_workers in VideoClips * add documentation for num_workers in VideoClips
-
Zhicheng Yan authored
* add _backend argument to __init__() of class VideoClips * minor fix * minor fix * Make backend private in VideoClips * Fix lint
-
Zhicheng Yan authored
* video transforms * [video transforms]in ToTensorVideo, divide value by 255.0 * [video transforms] fix a bug * fix linting * Make changes backwards-compatible
-
Ardalan authored
* add tar.xz archive handler * update unittest for tar.xz archive * remove .tar.xz unittest * add separate .tar.xz unittest * update PY2 compatibility
-
Francisco Massa authored
-
peterjc123 authored
-
- 23 Sep, 2019 4 commits
-
-
ekosman authored
-
Dmitry Nikulin authored
* Support empty target_type for CelebA dataset * Return (X, None) for interface consistency * Document behavior for target_type=[] * Simplify diff * Raise exception on meaningless parameters
-
Surgan Jandial authored
* format param added * lint fixes * lint fixes * lint fixes
-
Dmitry Belenko authored
* Add initial mnasnet impl * Remove all type hints, comply with PyTorch overall style * Expose models * Remove avgpool from features() and add separately * Fix python3-only stuff, replace subclasses with functions * fix __all__ * Fix typo * Remove conditional dropout * Make dropout functional * Addressing @fmassa's feedback, round 1 * Replaced adaptive avgpool with mean on H and W to prevent collapsing the batch dimension * Partially address feedback * YAPF * Removed redundant class vars * Update urls to releases * Add information to models.rst * Replace init with kaiming_normal_ in fan-out mode * Use load_state_dict_from_url * Fix depth scaling on first 2 layers * Restore initialization * Match reference implementation initialization for dense layer * Meant to use Kaiming * Remove spurious relu * Point to the newest 0.5 checkpoint * Latest pretrained checkpoint * Restore 1.0 checkpoint * YAPF * Implement backwards compat as suggested by Soumith * Update checkpoint URL * Move warnings up * Record a couple more function parameters * Update comment * Set the correct version such that if the BC-patched model is saved, it could be reloaded with BC patching again * Set a member var, not class var * Update mnasnet.py Remove unused member var as per review. * Update the path to weights
-
- 22 Sep, 2019 1 commit
-
-
peterjc123 authored
-
- 20 Sep, 2019 5 commits
-
-
Francisco Massa authored
-
Zhicheng Yan authored
* [video reader] inception commit * add method save_metadata to class VideoClips in video_utils.py * add load_metadata() method to VideoClips class * add Exception to not catch unexpected events such as memory erros, interrupt * fix bugs in video_plus.py * [video reader]remove logging. update setup.py * remove time measurement in test_video_reader.py * Remove glog and try making ffmpeg finding more robust * Add ffmpeg to conda build * Add ffmpeg to conda build [again] * Make library path finding more robust * Missing import * One more missing fix for import * Py2 compatibility and change package to av to avoid version conflict with ffmpeg * Fix for python2 * [video reader] support to decode one stream only (e.g. video/audio stream) * remove argument _precomputed_metadata_filepath * remove save_metadata method * add get_metadata method * expose _precomputed_metadata and frame_rate arguments in video dataset __init__ method * remove ssize_t * remove size_t to pass CI check on Windows * add PyInit__video_reader function to pass CI check on Windows * minor fix to define PyInit_video_reader symbol * Make c++ video reader optional * Temporarily revert changes to test_io * Revert changes to python files * Rename files to make it private * Fix python lint * Fix C++ lint * add a functor object EnumClassHash to make Enum class instances usable as key type of std::unordered_map * fix cpp format check
-
eellison authored
* script_fcn_resnet * Make old models load * DeepLabV3 also got torchscript-ready
-
Francisco Massa authored
* [TEST][DO NOT MERGE] Testing CI * Try fixing build
-
eellison authored
* make densenet scriptable * make py2 compat * use torch List polyfill * fix unpacking for checkpointing * fewer changes to _Denseblock * improve error message * print traceback * add typing dependency * add typing dependency to travis too * Make loading old checkpoints work
-
- 18 Sep, 2019 5 commits
-
-
Francisco Massa authored
* Remove C++ extensions in favor of custom ops * Remove unused custom_ops.cpp file * Rename _custom_ops.py * Reorganize functions * Minor improvements and fixes * Fix lint * Fully scriptable ops * Import types used by annotations
-
Francisco Massa authored
* Add Hub tests in torchvision * Run test_hub in its own interpreter
-
Francisco Massa authored
* Make AnchorGenerator support half precision * Add test for fasterrcnn with double * convert gt_boxes to right dtype
-
peterjc123 authored
* Fix Windows CI * Use correct version
-
peterjc123 authored
* Windows build scripts Add testing yaml * Change urls * Test MS agent * Switch base image * Use fb links
-
- 17 Sep, 2019 3 commits
-
-
Francisco Massa authored
* Set block style serialization in yaml * Add new checkout command * Put on a different branch name
-
Lara Haidar authored
* Support Exporting GeneralizedRCNNTransform * refactor code to address comments * update tests * address comments * revert min_size to test CI * re-revert min_size
-
eellison authored
-
- 16 Sep, 2019 1 commit
-
-
Francisco Massa authored
-
- 12 Sep, 2019 1 commit
-
-
Karl Ostmo authored
use flake8 with Python 3 on .circleci directory
-
- 11 Sep, 2019 1 commit
-
-
Philip Meier authored
* Adds optional fill colour to rotate * bug fix
-
- 10 Sep, 2019 3 commits
-
-
Thomas Viehmann authored
* Make custom ops differentiable and replace autograd.Function. Use ops unconditionally. We may consider removing the extension functions in a follow-up. The code-path is tested by the exisitng tests for differentiability. * add scripting gradchecks tests and use intlist * fix implicit tuple conversion for gcc-5 * fix merge
-
Dmitrii Petukhov authored
-
Lara Haidar authored
* Revert "Revert "Register Torchvision Ops as Cutom Ops (#1267)" (#1316)" This reverts commit fe234fc8. * Make import of C++ extensions lazy * define python initialization functions for extension * Fix lint
-
- 09 Sep, 2019 4 commits
-
-
Philip Meier authored
* fix url * update comment
-
Edward Z. Yang authored
This reverts commit 78f169b5.
-
Francisco Massa authored
* [WIP] Add CircleCI for CI * Make jobs only run on master * Add initial CI * [wip] testing if works * Trying some basic GPU tests * [WIP] maybe it will work? * [WIP] One more try * Pin versions * Simplify and reuse * Fix * [WIP] testing windows * [WIP] testing windows * Try windows * Try Windows * Try windows * Try windows * Try windows * Try windows * Try windows * Try windows * Windows speedup * Windows multicores * Add parallel flags for Windows * Skip some tests on Windows * Sync config.yml and config.yml.in * Regenerate * Run all tests * Limit python and cuda version for finding pytorch * Skip darwin for previous check * Add description * Fix logic * Remove space * Add CUDA test back * Add back .travis.yml for now and remove duplicate test * Add newline
-
Lara Haidar authored
* Register torchvision ops * install ORT only with python 3 * remane lib + address other comments * fix lint * fix lib copy * find file with pattern instead of suffix * use relative path * revert rename and use imp to find lib * fix typo
-
- 06 Sep, 2019 2 commits
-
-
Vishwak Srinivasan authored
-
Philip Meier authored
-