Josua Rieder 8d81fdf3d9 Fix typos

2025-01-19 13:39:40 -08:00

..

_info

…

readers

Fix typos

2025-01-19 13:39:40 -08:00

__init__.py

…

auto_augment.py

Fix typos

2025-01-19 13:39:40 -08:00

config.py

Add in_chans to data config helper. Fix #2021

2023-11-23 12:44:59 -08:00

constants.py

Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate

2022-12-05 10:21:34 -08:00

dataset_factory.py

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -08:00

dataset_info.py

…

dataset.py

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -08:00

distributed_sampler.py

…

imagenet_info.py

Fix typos

2025-01-19 13:39:40 -08:00

loader.py

Add half-precision (bfloat16, float16) support to train & validate scripts. Should push dtype handling into model factory / pretrained load at some point...

2025-01-07 10:25:14 -08:00

mixup.py

Replace deprecated NumPy aliases of builtin types

2023-07-03 22:24:25 +03:00

random_erasing.py

Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.

2022-09-23 16:08:59 -07:00

real_labels.py

…

tf_preprocessing.py

Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate

2022-12-05 10:21:34 -08:00

transforms_factory.py

Tweak a few docstrings

2024-11-13 10:12:31 -08:00

transforms.py

Add 'Maybe' PIL / image tensor conversions in case image alread in tensor format

2024-07-08 13:43:51 -07:00