- 08 Oct, 2019 3 commits
-
-
Zhicheng Yan authored
-
Zhicheng Yan authored
-
Francisco Massa authored
Revert "Change all torch::nn::init::Nonlinearity::{name} and torch::nn::init::FanMode::{name} to torch::k{name} (#1394)" (#1428) This reverts commit 8c3cea7f.
-
- 07 Oct, 2019 4 commits
-
-
Francisco Massa authored
-
Adam J. Stewart authored
* Fix DeprecationWarning for collections.Iterable import * Simplify version comparison
-
Jaesun Park authored
* Fix hmdb51.py typo * Fix ucf101.py typo
-
Zhicheng Yan authored
-
- 04 Oct, 2019 3 commits
-
-
Lara Haidar authored
* Support Exporting MultiScaleRoiAlign to ONNX * remove cast * fix dtype * move cast
-
Zhicheng Yan authored
* move sampler into TV core. Update UniformClipSampler * Fix reference training script * Skip test if pyav not available * change interpolation from round() to floor() as round(0.5) behaves differently between py2 and py3
-
Koen van de Sande authored
Fix reference training script for Mask R-CNN for PyTorch 1.2 (during evaluation after epoch, mask datatype became bool, pycocotools expects uint8) (#1413)
-
- 03 Oct, 2019 2 commits
-
-
Zhicheng Yan authored
* add metadata to video dataset classes. bug fix. more robustness * query video backend within VideoClips class * Fix tests * Fix lint
-
Francisco Massa authored
* Temporary fix for CI * Missing install
-
- 02 Oct, 2019 3 commits
-
-
eellison authored
* add support for video resnet models, restructure script test to just ignore RCNN models * switch back to testing subset of the models
-
Stefan Otte authored
-
Will Feng authored
Change all torch::nn::init::Nonlinearity::{name} and torch::nn::init::FanMode::{name} to torch::k{name} (#1394) * Change all torch::nn::init::Nonlinearity::{name} and torch::nn::init::FanMode::{name} to torch::k{name} * empty commit * fix lint * fix lint * fix lint
-
- 01 Oct, 2019 2 commits
-
-
eellison authored
* add expected result tests * fix wrong assertion * start with only detection models * remove unneeded rng setting * fix test * add tuple support * update test * syntax error * treat .pkl files as binary data, see : https://git-scm.com/book/en/v2/Customizing-Git-Git-Attributes#_binary_files * fix test * fix elif * Map tensor results and enforce maximum pickle size * unrelated change * larger rtol * pass rtol atol around * last commit i swear... * respond to comments * fix flake * fix py2 flake
-
Xiuyan Ni authored
-
- 30 Sep, 2019 4 commits
-
-
Will Feng authored
Change all torch::nn::init::Nonlinearity::{name} and torch::nn::init::FanMode::{name} usage to torch::k{name} -
Chandresh Kanani authored
modified code of io.read_video and io.read_video_timestamps to intepret pts values in seconds (#1331) * modified code of io.read_video and io.read_video_timestamps to interpret pts values in seconds * changed default value for pts_unit to pts, corrected formatting * hanndliing both fractions and floats for start_pts and end_pts, added test cases for pts_unit sec * moved unit conversion logic to _read_from_stream method
-
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
-