GitLab

* fix MNIST byte flipping

* add test

* move to utils

* remove lazy import

* fix typos throughout the code base

* fix grammar

* revert formatting changes to gallery

* revert 'an uXX'

* remove 'number of the best'

* _weight_size -> _file_size

* Better formatting of individual weights tables

* Remove file size from main tables to avoid confusion with weight size (as in RAM)

* Remove unnecessary (file size) suffix

* Fix CI error?

* Formatting
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* [WIP] Remove torch.jit.fuser("fuser2") in test

Internally we're considering removing support for fuser2, so we need to remove this special case from the test.

* completely remove special-casing

* np.int -> np.int32

* Update test/test_transforms.py

* add moving mnist dataset
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* remove unused modules
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* modify docstring
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* modify docstring and docs
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* add split and split ratio kwargs
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* fix checking split argument
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* remove unused package

* delete lines
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* fix filename property
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* fix reviews
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* modify docstrings
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* add split tests and etc
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* fix tests
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>
Signed-off-by: tsugumi-sys <tidemark0105@gmail.com>

* improve prototype GPU CI config

* consolidate CPU and GPU prototype workflows

* [REVERTME] deactivate other CI

* syntax

* fix name

* fix echo mutex

* style

* fix runner

* add jobname

* run all prototype tests

* fix mutex echo

* cleanup

* Revert "[REVERTME] deactivate other CI"

This reverts commit c931ff2c27bbc2b04324b777ae39a6fc25e50b8a.

* remove old workflows

* use free GHA runner for CPU tests

* Revert "use free GHA runner for CPU tests"

This reverts commit a209b31387d9d14bc6446743587db244bccc2106.

* [REVERTME] turn warnings into errors

* try fix test warnings

* fix module import in tests

* revert

* int -> float

* [REVERTME] why do we have to mark as safe?

* Try add GPU ci for prototype transforms

* Update prototype-transforms-tests-linux-gpu.yml

* Update prototype-transforms-tests-linux-gpu.yml

* Update prototype-transforms-tests-linux-gpu.yml

* Update prototype-transforms-tests-linux-gpu.yml

* Update prototype-transforms-tests-linux-gpu.yml

* add small tolerance for adjust_contrast on CUDA

* [skip-ci] update
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* Add cuda resnet50 test

* Fix path

* Tune vision smoke test

* Add more advanced smoke test

* add torch import

* remove dependency on torch

* Add missing vars

* More code and ufmt
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* improve download tests

* move SBU to xfailed

* add SSL version output to CI

* fix prototype dataset imagenet mock data

* [REVERT ME] temporarily add prototype datasets test to CI

* Revert "[REVERT ME] temporarily add prototype datasets test to CI"

This reverts commit 7cafb5f01487c7625bcffd88e76832349b3e5055.

* rename features._Feature to datapoints.Datapoint

* _Datapoint to Datapoint

* move is_simple_tensor to transforms.utils

* fix CI

* move Datapoint out of public namespace

* Adding _ops and _weight_size metadata checks to tests

* Fixing wrong ops value

* Changing test_schema_meta_validation to instantiate the model only once

* moving instantiating quantized models inside get_ops

* Small refactor of test_schema_meta_validation logic

* Reverting to previous ops value

* Simplifying unquantized models logic in test_schema_meta_validation

* Update test/test_extended_models.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Toni Blaslov <tblaslov@fb.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* expose some prototype transforms utils

* rename _isinstance

* Fix extensions in wheels build

* pass pre-script from caller

* add pre-script to run relocate on wheel

* install auditwheel for relocate

* reorder numpy install

* use jpeg80 for py3.9 and py3.10 mac builds

* revert back to jpeg9

* printing the found jpeg libs during build time and reintro circle for comparison

* more debug logs, remove extra linux builds

* typo fix

* brew uninstall conflicting jpeg libraries on mac machines

* fix path for unittests

* ufmt

* introduce type failures in dispatchers

* add type checks to all dispatchers

* add missing else

* add test

* fix convert_color_space

* minor cleanup of the prototype transforms tests

* refactor ImagePair

* pretty format enum

* use non-random images for interpolation kernels for testing

* use real image rather than artificial

* cleanup

* update tests

* revert replacing expect

* test_detection_model freeze_rng_state

* update tests

* Test: add backend parameter

* VideoReader object now works on backend

* Frame reading now passes

* Keyframe seek now passes

* Pyav backend now supports metadata

* changes in test to reflect GPU decoder change

* Linter?

* Test GPU output

* Addressing Joao's comments

* lint

* lint

* Revert "Test GPU output"

This reverts commit f62e955d7dc81bcb23b40d58ea75413b9b62e76d.

* lint?

* lint

* lint

* Address issues in build?

* hopefully doc fix

* Arrgh

* arrgh

* fix typos

* fix input options

* remove read from memory option in pyav

* skip read from mem test for gpu and pyab be

* fix test

* remove unused import

* Hack to get reading from memory work with pyav

* patch audio test

* gallery change in a hope that docs won't break

* check video decoder inside io

* adding missing lib loading code

* remove unused input
Co-authored-by: Bruno Korbar <bkorbar@quansight.com>
Co-authored-by: Joao Gomes <jdsgomes@fb.com>

* Just start adding mere copy paste

* Replace d with t and D with T

* Align swin transformer video to image a bit

* Rename d -> t

* align with 2d impl

* align with 2d impl

* Add helpful comments and config for 3d

* add docs

* Add docs

* Add configurations

* Add docs

* Fix bugs

* Fix wrong edit

* Fix wrong edit

* Fix bugs

* Fix bugs

* Fix as per fx suggestions

* Update torchvision/models/video/swin_transformer.py

* Fix as per fx suggestions

* Fix expect files and code

* Update the expect files

* Modify video swin

* Add min size and min temporal size, num params

* Add flops and size

* Fix types

* Fix url recipe
Co-authored-by: Yosua Michael Maranatha <yosuamichael@fb.com>

* Clean up `_get_inverse_affine_matrix` and `_compute_affine_output_size`

* Optimize `_apply_grid_transform`

* Cleanup `_assert_grid_transform_inputs`

* Fix bugs on `_pad_with_scalar_fill` & `crop_mask` and port `crop_image_tensor`

* Call directly `_pad_with_scalar_fill`

* Fix linter

* Clean up `center_crop_image_tensor`

* Fix comments.

* Fixing rounding issues.

* Bumping tolerance for rotate which is unrelated to this PR.

* Fix tolerance threshold for RandomPerspective.

* Clean up `_affine_grid` and `affine_image_tensor`

* Clean up `rotate_image_tensor`

* Fixing linter

* Address code-review comments.

* Adding FLOPs and size to model metadata

* Adding weight size to quantization models

* Small refactor of rich metadata

* Removing unused code

* Fixing wrong entries

* Adding .DS_Store to gitignore

* Renaming _flops to _ops

* Adding number of operations to quantization models

* Reflecting _flops change to _ops

* Renamed ops and weight size in individual model doc pages

* Linter fixes

* Rounding ops to first decimal

* Rounding num ops and sizes to 3 decimals

* Change naming of columns.

* Update tables
Co-authored-by: Toni Blaslov <tblaslov@fb.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>