LICENSE
NOTICE
README.md
pyproject.toml
src/torch_batteries/__init__.py
src/torch_batteries/const.py
src/torch_batteries/py.typed
src/torch_batteries.egg-info/PKG-INFO
src/torch_batteries.egg-info/SOURCES.txt
src/torch_batteries.egg-info/dependency_links.txt
src/torch_batteries.egg-info/requires.txt
src/torch_batteries.egg-info/top_level.txt
src/torch_batteries/callbacks/__init__.py
src/torch_batteries/callbacks/_model_checkpoint_functions.py
src/torch_batteries/callbacks/_monitor.py
src/torch_batteries/callbacks/base.py
src/torch_batteries/callbacks/early_stopping.py
src/torch_batteries/callbacks/experiment_tracking.py
src/torch_batteries/callbacks/gradient_accumulation.py
src/torch_batteries/callbacks/gradient_clip.py
src/torch_batteries/callbacks/learning_rate_scheduler.py
src/torch_batteries/callbacks/mixed_precision.py
src/torch_batteries/callbacks/model_checkpoint.py
src/torch_batteries/callbacks/terminate_on_non_finite.py
src/torch_batteries/data/__init__.py
src/torch_batteries/data/base.py
src/torch_batteries/data/handler.py
src/torch_batteries/data/loader.py
src/torch_batteries/data/types/__init__.py
src/torch_batteries/data/types/aliases.py
src/torch_batteries/data/types/data_context.py
src/torch_batteries/data/types/data_loader_bundle.py
src/torch_batteries/data/types/data_loader_config.py
src/torch_batteries/data/types/dataset_bundle.py
src/torch_batteries/data/types/resolved_data.py
src/torch_batteries/events/__init__.py
src/torch_batteries/events/_handler_base.py
src/torch_batteries/events/_metadata.py
src/torch_batteries/events/handler.py
src/torch_batteries/events/core/__init__.py
src/torch_batteries/events/core/charge.py
src/torch_batteries/events/core/event.py
src/torch_batteries/events/core/event_context.py
src/torch_batteries/events/core/optimization_step.py
src/torch_batteries/tracking/__init__.py
src/torch_batteries/tracking/base.py
src/torch_batteries/tracking/types.py
src/torch_batteries/tracking/wandb/__init__.py
src/torch_batteries/tracking/wandb/_artifact.py
src/torch_batteries/tracking/wandb/_module.py
src/torch_batteries/tracking/wandb/_run.py
src/torch_batteries/tracking/wandb/tracker.py
src/torch_batteries/trainer/__init__.py
src/torch_batteries/trainer/_checkpoint.py
src/torch_batteries/trainer/_evaluation.py
src/torch_batteries/trainer/_prediction.py
src/torch_batteries/trainer/_state.py
src/torch_batteries/trainer/_training.py
src/torch_batteries/trainer/context.py
src/torch_batteries/trainer/core.py
src/torch_batteries/trainer/types/__init__.py
src/torch_batteries/trainer/types/fit_result.py
src/torch_batteries/trainer/types/predict_result.py
src/torch_batteries/trainer/types/step_output.py
src/torch_batteries/trainer/types/test_result.py
src/torch_batteries/trainer/types/train_result.py
src/torch_batteries/trainer/types/validation_result.py
src/torch_batteries/utils/__init__.py
src/torch_batteries/utils/batch.py
src/torch_batteries/utils/device.py
src/torch_batteries/utils/formatting.py
src/torch_batteries/utils/prediction.py
src/torch_batteries/utils/logging/__init__.py
src/torch_batteries/utils/logging/_functions.py
src/torch_batteries/utils/logging/_manager.py
src/torch_batteries/utils/metrics/__init__.py
src/torch_batteries/utils/metrics/_helpers.py
src/torch_batteries/utils/metrics/calculate.py
src/torch_batteries/utils/metrics/metric_types.py
src/torch_batteries/utils/metrics/phase_manager.py
src/torch_batteries/utils/metrics/state/__init__.py
src/torch_batteries/utils/metrics/state/base.py
src/torch_batteries/utils/metrics/state/collected.py
src/torch_batteries/utils/progress/__init__.py
src/torch_batteries/utils/progress/base.py
src/torch_batteries/utils/progress/factory.py
src/torch_batteries/utils/progress/progress_bar.py
src/torch_batteries/utils/progress/silent.py
src/torch_batteries/utils/progress/simple.py
src/torch_batteries/utils/progress/types/__init__.py
src/torch_batteries/utils/progress/types/phase.py
src/torch_batteries/utils/progress/types/progress_metrics.py