GitLab

* 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

* 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

* fixed typo

* fixed some more typos and grammer

* Fix flakiness of test_randomresized_params

* Real fix

* Reduce number of iters

* Added initial code

* Changed clang version

* Changed travis script

* Changed travis script

* Updated travis script

* Updated travis script

* Updated travis script

* Use Tensor.data_ptr instead of .data

* use pytorch-nightly in CI

* Fix bug of changing input tensor in function

The original code will change the image tensor passed in by multiplying it by 255. The PR fixes that.

* fix inplace multiplication in save_image

* test that torchhub models are scriptable

* fix lint

Fixed video labelling after `subset` call for HMDB51 dataset (hmdb51.py) (EDIT: UCF101 as well) (#1240)

* Fixed video labelling after subset for HMDB51 dataset

* Fixed video labelling after subset for HMDB51 dataset
Co-authored-by: Eric Tang <etang21@stanford.edu>
Co-authored-by: Ryan Cao <ryancao@stanford.edu>

* UCF 101 Labeling fixes

- Analogous fix to HMDB51 to maintain correct labels after the train-test split
- Additional change to the `select_fold` method in `ucf101.py` to correctly reflect the annotation format
Co-authored-by: Ryan Cao <ryancao@stanford.edu>
Co-authored-by: Eric Tang <etang21@stanford.edu>

* explain lr and batch size in references/detection/train.py

* fix typo

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Actually upload wheels (please port to master)
Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Propagate more environment variables.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Put macos binaries in the right place
Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Add uploading support, make CUDA builds actually work.

Following a similar line of inquiry to pytorch/audio#217

`self.fc1(x)` converts the shape of `x` into "N x 1024", and `self.fc2(x)` converts into "N x num_classes".

By adding `print(x.shape)` under each comment line, the console displays as follows (batch_size is 1):

```text
torch.Size([1, 2048])
torch.Size([1, 1024])
torch.Size([1, 1024])
torch.Size([1, 1024])
torch.Size([1, 1000])
```

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Add docs for video models

* Fix docstrings for resnet and vgg

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* [WIP] Minor cleanups on R3d

* Move all models to video/resnet.py

* Remove old files

* Make tests less memory intensive

* Lint

* Fix typo and add pretraing arg to training script

* Expose docs for io and ops package

Had do modify the docstrings to use Napoleon NumPy style, because Napoleon Google Style doesn't support multiple return arguments

* Add video section

* Add docs to UCF101 and HMDB51 datasets

* Remove extra line

* Lint

* Move RandomClipSampler to references

* Lint and bugfix

Also add docs

* Copy classification scripts for video classification

* Initial version of video classification

* add version

* Training of r2plus1d_18 on kinetics work

Gives even slightly better results than expected, with 57.336 top1 clip accuracy. But we count some clips twice in this evaluation

* Cleanups on training script

* Lint

* Minor improvements

* Remove some hacks

* Lint

* Optimize read_video_timestamps for some formats

* Add some tests