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
74d2829341
Merge branch 'master' into bits_and_tpu
|
4 years ago |
| .. |
|
parsers
|
Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.
|
4 years ago |
|
__init__.py
|
More dataset work including factories and a tensorflow datasets (TFDS) wrapper
|
4 years ago |
|
auto_augment.py
|
Enable uniform augmentation magnitude sampling and set AugMix default
|
4 years ago |
|
collate.py
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
|
config.py
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
|
constants.py
|
…
|
|
|
dataset.py
|
Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.
|
4 years ago |
|
dataset_factory.py
|
Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.
|
4 years ago |
|
distributed_sampler.py
|
…
|
|
|
fetcher.py
|
Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.
|
4 years ago |
|
loader.py
|
Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.
|
4 years ago |
|
mixup.py
|
…
|
|
|
prefetcher_cuda.py
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
|
random_erasing.py
|
…
|
|
|
real_labels.py
|
…
|
|
|
tf_preprocessing.py
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
|
transforms.py
|
Support for huggingface hub via create_model and default_cfgs.
|
4 years ago |
|
transforms_factory.py
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
