LICENSE
README.md
pyproject.toml
setup.cfg
irisml/tasks/append_classifier.py
irisml/tasks/benchmark_model.py
irisml/tasks/build_classification_prompt_dataset.py
irisml/tasks/build_zero_shot_classifier.py
irisml/tasks/create_classification_prompt_generator.py
irisml/tasks/evaluate_accuracy.py
irisml/tasks/evaluate_detection_average_precision.py
irisml/tasks/export_onnx.py
irisml/tasks/get_targets_from_dataset.py
irisml/tasks/make_feature_extractor_model.py
irisml/tasks/make_image_text_contrastive_model.py
irisml/tasks/make_image_text_transform.py
irisml/tasks/make_oversampled_dataset.py
irisml/tasks/predict.py
irisml/tasks/split_image_text_model.py
irisml/tasks/train_with_gradient_cache.py
irisml/tasks/train/__init__.py
irisml/tasks/train/build_dataloader.py
irisml/tasks/train/build_lr_scheduler.py
irisml/tasks/train/build_optimizer.py
irisml/tasks/train/ddp_trainer.py
irisml/tasks/train/ddp_utils.py
irisml/tasks/train/plugin_list.py
irisml/tasks/train/task.py
irisml/tasks/train/trainer.py
irisml/tasks/train/plugins/amp.py
irisml/tasks/train/plugins/clip_grad_norm.py
irisml/tasks/train/plugins/ddp.py
irisml/tasks/train/plugins/ema.py
irisml/tasks/train/plugins/log_summary.py
irisml/tasks/train/plugins/log_tensorboard.py
irisml/tasks/train/plugins/plugin_base.py
irisml/tasks/train/plugins/progressbar.py
irisml/tasks/train/plugins/scale_loss.py
irisml_tasks_training.egg-info/PKG-INFO
irisml_tasks_training.egg-info/SOURCES.txt
irisml_tasks_training.egg-info/dependency_links.txt
irisml_tasks_training.egg-info/requires.txt
irisml_tasks_training.egg-info/top_level.txt
test/test_append_classifier.py
test/test_benchmark_model.py
test/test_build_classification_prompt_dataset.py
test/test_build_zero_shot_classifier.py
test/test_create_classification_prompt_generator.py
test/test_dataloader.py
test/test_ddp_trainer.py
test/test_evaluate_accuracy.py
test/test_evaluate_detection_average_precision.py
test/test_export_onnx.py
test/test_get_targets_from_dataset.py
test/test_make_feature_extractor_model.py
test/test_make_image_text_contrastive_model.py
test/test_make_image_text_transform.py
test/test_make_oversampled_dataset.py
test/test_predict.py
test/test_split_image_text_model.py
test/test_train.py
test/test_train_with_gradient_cache.py