Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs. ...

* improve consistency of model creation helper fns
* add comments to some of the model helpers
* support passing external default_cfgs so they can be sourced from hub

2021-03-16 22:48:26 -07:00

..

parsers

Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py

2021-01-22 16:54:33 -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 some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

config.py

Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg.

2021-02-06 16:30:02 -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

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

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

dataset.py

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

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

distributed_sampler.py

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

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

loader.py

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

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

mixup.py

More flexible mixup mode, add 'half' mode.

2020-09-07 20:03:06 -07:00

random_erasing.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07: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

Support for huggingface hub via create_model and default_cfgs.

2021-03-16 22:48:26 -07:00

transforms.py

Support for huggingface hub via create_model and default_cfgs.

2021-03-16 22:48:26 -07:00