Ross Wightman f5b58e31a2 Allow non train mode for wds reader to operate w/o sample count, exhaust iterator

2024-10-31 16:39:35 -07:00

..

_info

…

readers

Allow non train mode for wds reader to operate w/o sample count, exhaust iterator

2024-10-31 16:39:35 -07: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

…

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

…

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

Fix pass through of input / target keys so ImageDataset readers so args work with hfds instead of just hfids (iterable)

2024-07-17 10:11:46 -07:00

distributed_sampler.py

Significant model refactor and additions:

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

imagenet_info.py

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

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

loader.py

fix device check

2024-10-17 12:38:02 +00:00

mixup.py

…

random_erasing.py

…

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

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

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

transforms.py

…