CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
examples/1-basic.py
examples/10-multi-optimizers.py
examples/11-multi-dataloaders.py
examples/12-node-classification.py
examples/13-weight-grad-visualize.py
examples/14-costomize-callback.py
examples/15-customize-steps-1.py
examples/16-customize-steps-2.py
examples/17-costomize-steps-3.py
examples/18-use_patches-1.py
examples/19-use_patches-2.py
examples/2-basic-params.py
examples/20-customize-patch.py
examples/21-distributed.py
examples/22-grad_accumulate-1.py
examples/23-grad_accumulate-2.py
examples/24-dict_inputs_outputs.py
examples/3-metrics.py
examples/4-checkpoint-earlystop.py
examples/5-lr-find.py
examples/6-logger.py
examples/7-log-hyperparameters.py
examples/8-interprete.py
examples/9-lr-schedule.py
imgs/data_flow.png
src/deepepochs/__init__.py
src/deepepochs/distributed.py
src/deepepochs/loops.py
src/deepepochs/metrics.py
src/deepepochs/optimizer.py
src/deepepochs/patches.py
src/deepepochs/tools.py
src/deepepochs.egg-info/PKG-INFO
src/deepepochs.egg-info/SOURCES.txt
src/deepepochs.egg-info/dependency_links.txt
src/deepepochs.egg-info/requires.txt
src/deepepochs.egg-info/top_level.txt
src/deepepochs/callbacks/__init__.py
src/deepepochs/callbacks/analyze.py
src/deepepochs/callbacks/callback.py
src/deepepochs/callbacks/check.py
src/deepepochs/callbacks/default.py
src/deepepochs/callbacks/grad_clip.py
src/deepepochs/callbacks/interprete.py
src/deepepochs/callbacks/log.py
src/deepepochs/callbacks/lr_find.py
src/deepepochs/trainer/__init__.py
src/deepepochs/trainer/epoch_task.py
src/deepepochs/trainer/loss_wrapper.py
src/deepepochs/trainer/model_wrapper.py
src/deepepochs/trainer/trainer.py
test_results/rtx3080/benchmark-1-bf16.json
test_results/rtx3080/benchmark-1-fp16.json
test_results/rtx3080/benchmark-1-fp32.json
test_results/rtx3080/benchmark-2-bf16.json
test_results/rtx3080/benchmark-2-fp16.json
test_results/rtx3080/benchmark-2-fp32.json
test_results/rtx3080/benchmark-4-bf16.json
test_results/rtx3080/benchmark-4-fp16.json
test_results/rtx3080/benchmark-4-fp32.json
test_results/rtx3080/environment.json
test_results/rtx3080/final-runs.json
test_results/rtx3080/integration.json
test_results/rtx3080/summary.json
test_results/rtx3080/training-2-bf16.json
test_results/rtx3080/training-2-fp16.json
test_results/rtx3080/training-2-fp32.json
test_results/rtx3080/training-4-bf16.json
test_results/rtx3080/training-4-fp16.json
test_results/rtx3080/training-4-fp32.json
test_results/rtx3080/unit-tests.json
test_results/rtx3080/evaluation-2-bf16/evaluation-2gpu-bf16.json
test_results/rtx3080/evaluation-2-fp16/evaluation-2gpu-fp16.json
test_results/rtx3080/evaluation-2-fp32/evaluation-2gpu-fp32.json
test_results/rtx3080/evaluation-4-bf16/evaluation-4gpu-bf16.json
test_results/rtx3080/evaluation-4-fp16/evaluation-4gpu-fp16.json
test_results/rtx3080/evaluation-4-fp32/evaluation-4gpu-fp32.json
test_results/rtx3080/single-gpu-portability/single-gpu-checkpoint.json
test_results/rtx3080/single-gpu-portability-fp16/single-gpu-checkpoint.json
tests/multigpu_benchmark.py
tests/multigpu_evaluation.py
tests/multigpu_training.py
tests/single_gpu_checkpoint.py
tests/smoke_training.py
tests/test_amp_clipping.py
tests/test_callbacks.py
tests/test_checkpoint.py
tests/test_checkpoint_current.py
tests/test_checkpoint_review.py
tests/test_data_review.py
tests/test_ddp_review.py
tests/test_distributed_context.py
tests/test_distributed_training.py
tests/test_gpu.py
tests/test_interprete_review.py
tests/test_loss_windows.py
tests/test_metrics.py
tests/test_packaging.py
tests/test_patches.py
tests/test_tools.py
tests/test_trainer.py