Junming Chen 569d114ef7

Fix device problem ...

Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.

2022-04-19 11:53:18 +08:00

..

parsers

Add commented code to increase open file limit via Python (for TFDS dataset building)

2021-11-30 17:02:27 -08:00

__init__.py

More dataset work including factories and a tensorflow datasets (TFDS) wrapper

2021-01-15 17:26:20 -08:00

auto_augment.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

config.py

Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.

2021-02-23 17:31:42 -08:00

constants.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

dataset_factory.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

dataset.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

distributed_sampler.py

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

loader.py

Significant model refactor and additions:

2022-02-28 13:56:23 -08:00

mixup.py

Fix device problem

2022-04-19 11:53:18 +08:00

random_erasing.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

real_labels.py

Fix a bug with accuracy retrieving from RealLabels

2020-12-04 16:12:50 +01:00

tf_preprocessing.py

Remove some old __future__ imports

2021-01-03 11:24:16 -08:00

transforms_factory.py

Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.

2022-01-14 16:28:27 -08:00

transforms.py

Add interpolation mode handling to transforms. Removes InterpolationMode warning. Works for torchvision versions w/ and w/o InterpolationMode enum. Fix #738.

2021-10-28 17:35:01 -07:00