Ross Wightman 7d121ac2ef Small tweak of timm ToTensor for clarity

2024-02-10 14:57:40 -08:00

..

_info

Add Microsoft FocalNet specific ('ms') ImageNet-22k classifier layout

2023-03-18 14:57:34 -07:00

readers

Fiddling with iterator wrapping for HF ds streaming

2024-01-09 12:41:54 -08:00

__init__.py

Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg

2023-02-08 08:29:20 -08:00

auto_augment.py

Fix #1713 missed assignement in 3-aug level fn, fix few other minor lint complaints in auto_augment.py

2023-03-11 14:32:23 -08:00

config.py

…

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

…

dataset_info.py

Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg

2023-02-08 08:29:20 -08:00

dataset.py

…

distributed_sampler.py

…

imagenet_info.py

…

loader.py

Pass train-crop-mode to create_loader/transforms from train.py args

2024-01-24 16:19:02 -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

Move ImageNet metadata (aka info) files to timm/data/_info. Add helper classes to make info available for labelling. Update inference.py for first use.

2023-02-06 17:45:03 -08:00

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

Pass train-crop-mode to create_loader/transforms from train.py args

2024-01-24 16:19:02 -08:00

transforms.py

Small tweak of timm ToTensor for clarity

2024-02-10 14:57:40 -08:00