Fredo Guan 81ca323751 Davit update formatting and fix grad checkpointing (#7) ... fixed head to gap->norm->fc as per convnext, along with option for norm->gap->fc failed tests due to clip convnext models, davit tests passed		2023-01-15 14:34:56 -08:00
..
readers	Davit update formatting and fix grad checkpointing (#7)	2023-01-15 14:34:56 -08:00
__init__.py	Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc	2022-10-14 15:14:38 -07:00
auto_augment.py	Davit update formatting and fix grad checkpointing (#7)	2023-01-15 14:34:56 -08:00
config.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
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.py	Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc	2022-10-14 15:14:38 -07:00
dataset_factory.py	Davit update formatting and fix grad checkpointing (#7)	2023-01-15 14:34:56 -08:00
distributed_sampler.py	…
loader.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
mixup.py	…
random_erasing.py	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2022-09-23 16:08:59 -07:00
real_labels.py	…
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.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 3-Augment support to auto_augment.py, clean up weighted choice handling, and allow adjust per op prob via arg string	2022-12-10 16:25:50 -08:00