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
d5aa17e415
Remove print from auto_augment
|
2 years ago |
| .. |
|
readers
|
Support HF datasets and TFSD w/ a sub-path by fixing split, fix #1598 ... add class mapping support to HF datasets in case class label isn't in info.
|
2 years ago |
|
__init__.py
|
Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc
|
2 years ago |
|
auto_augment.py
|
Remove print from auto_augment
|
2 years ago |
|
config.py
|
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
|
2 years ago |
|
constants.py
|
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
|
2 years ago |
|
dataset.py
|
Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc
|
2 years ago |
|
dataset_factory.py
|
Support HF datasets and TFSD w/ a sub-path by fixing split, fix #1598 ... add class mapping support to HF datasets in case class label isn't in info.
|
2 years ago |
|
distributed_sampler.py
|
Significant model refactor and additions:
|
3 years ago |
|
loader.py
|
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
|
2 years ago |
|
mixup.py
|
Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.
|
3 years ago |
|
random_erasing.py
|
Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.
|
2 years ago |
|
real_labels.py
|
Fix a bug with accuracy retrieving from RealLabels
|
4 years ago |
|
tf_preprocessing.py
|
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
|
2 years ago |
|
transforms.py
|
Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate
|
2 years ago |
|
transforms_factory.py
|
Add 3-Augment support to auto_augment.py, clean up weighted choice handling, and allow adjust per op prob via arg string
|
2 years ago |
