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_factory.py

Davit update formatting and fix grad checkpointing (#7)

2023-01-15 14:34:56 -08:00

dataset.py

Rename dataset/parsers -> dataset/readers, create_parser to create_reader, etc

2022-10-14 15:14:38 -07: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_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

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