You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 372ad5fa0d Significant model refactor and additions: ... * All models updated with revised foward_features / forward_head interface * Vision transformer and MLP based models consistently output sequence from forward_features (pooling or token selection considered part of 'head') * WIP param grouping interface to allow consistent grouping of parameters for layer-wise decay across all model types * Add gradient checkpointing support to a significant % of models, especially popular architectures * Formatting and interface consistency improvements across models * layer-wise LR decay impl part of optimizer factory w/ scale support in scheduler * Poolformer and Volo architectures added		3 years ago
..
parsers	Add commented code to increase open file limit via Python (for TFDS dataset building)	3 years ago
__init__.py	…
auto_augment.py	…
config.py	…
constants.py	…
dataset.py	…
dataset_factory.py	…
distributed_sampler.py	Significant model refactor and additions:	3 years ago
loader.py	Significant model refactor and additions:	3 years ago
mixup.py	…
random_erasing.py	…
real_labels.py	…
tf_preprocessing.py	…
transforms.py	…
transforms_factory.py	…