LICENSE
README.md
pyproject.toml
setup.cfg
irisml/tasks/append_classifier.py
irisml/tasks/benchmark_dataset.py
irisml/tasks/benchmark_model.py
irisml/tasks/benchmark_model_with_grad_cache.py
irisml/tasks/build_classification_prompt_dataset.py
irisml/tasks/build_zero_shot_classifier.py
irisml/tasks/concatenate_datasets.py
irisml/tasks/create_classification_prompt_generator.py
irisml/tasks/evaluate_accuracy.py
irisml/tasks/evaluate_detection_average_precision.py
irisml/tasks/exclude_negative_samples_from_classification_dataset.py
irisml/tasks/export_coco_from_detection_predictions.py
irisml/tasks/export_coco_from_torch_dataset.py
irisml/tasks/export_onnx.py
irisml/tasks/find_incorrect_classification_indices.py
irisml/tasks/find_incorrect_classification_multilabel_indices.py
irisml/tasks/get_subclass_dataset.py
irisml/tasks/get_targets_from_dataset.py
irisml/tasks/load_simple_classification_dataset.py
irisml/tasks/make_classification_dataset_from_object_detection.py
irisml/tasks/make_classification_dataset_from_predictions.py
irisml/tasks/make_detection_dataset_from_predictions.py
irisml/tasks/make_feature_extractor_model.py
irisml/tasks/make_fixed_prompt_image_transform.py
irisml/tasks/make_fixed_text_dataset.py
irisml/tasks/make_image_text_contrastive_model.py
irisml/tasks/make_image_text_transform.py
irisml/tasks/make_oversampled_dataset.py
irisml/tasks/make_prompt_list_image_transform.py
irisml/tasks/map_classification_predictions_to_detection.py
irisml/tasks/num_iters_to_epochs.py
irisml/tasks/predict.py
irisml/tasks/remove_empty_images_from_dataset.py
irisml/tasks/sample_few_shot_dataset.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/plugin_loader.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/compile.py
irisml/tasks/train/plugins/ddp.py
irisml/tasks/train/plugins/detect_anomaly.py
irisml/tasks/train/plugins/early_stopping.py
irisml/tasks/train/plugins/ema.py
irisml/tasks/train/plugins/fp16.py
irisml/tasks/train/plugins/freeze_modules.py
irisml/tasks/train/plugins/freeze_modules_upto.py
irisml/tasks/train/plugins/heartbeat.py
irisml/tasks/train/plugins/log_summary.py
irisml/tasks/train/plugins/log_tensorboard.py
irisml/tasks/train/plugins/mixup.py
irisml/tasks/train/plugins/nop.py
irisml/tasks/train/plugins/plugin_base.py
irisml/tasks/train/plugins/progressbar.py
irisml/tasks/train/plugins/scale_loss.py
irisml/tasks/train/plugins/step_decay_ema.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_dataset.py
test/test_benchmark_model.py
test/test_build_classification_prompt_dataset.py
test/test_build_zero_shot_classifier.py
test/test_concatenate_datasets.py
test/test_create_classification_prompt_generator.py
test/test_ddp_trainer.py
test/test_evaluate_accuracy.py
test/test_evaluate_detection_average_precision.py
test/test_exclude_negative_samples_from_classification_dataset.py
test/test_export_coco_from_detection_predictions.py
test/test_export_coco_from_torch_dataset.py
test/test_export_onnx.py
test/test_find_incorrect_classification_indices.py
test/test_find_incorrect_classification_multilabel_indices.py
test/test_get_subclass_dataset.py
test/test_get_targets_from_dataset.py
test/test_irisml.py
test/test_load_simple_classification_dataset.py
test/test_make_classification_dataset_from_object_detection.py
test/test_make_classification_dataset_from_predictions.py
test/test_make_detection_dataset_from_predictions.py
test/test_make_feature_extractor_model.py
test/test_make_fixed_prompt_image_transform.py
test/test_make_fixed_text_dataset.py
test/test_make_image_text_contrastive_model.py
test/test_make_image_text_transform.py
test/test_make_oversampled_dataset.py
test/test_make_prompt_list_image_transform.py
test/test_map_classification_predictions_to_detection.py
test/test_num_iters_to_epochs.py
test/test_predict.py
test/test_remove_empty_images_from_dataset.py
test/test_sample_few_shot_dataset.py
test/test_split_image_text_model.py
test/test_train_with_gradient_cache.py