LICENCE
README.md
README.rst
setup.py
basetrainer/__init__.py
basetrainer.egg-info/PKG-INFO
basetrainer.egg-info/SOURCES.txt
basetrainer.egg-info/dependency_links.txt
basetrainer.egg-info/not-zip-safe
basetrainer.egg-info/top_level.txt
basetrainer/callbacks/__init__.py
basetrainer/callbacks/callbacks.py
basetrainer/callbacks/log_history.py
basetrainer/callbacks/losses_recorder.py
basetrainer/callbacks/model_checkpoint.py
basetrainer/callbacks/multi_losses_recorder.py
basetrainer/criterion/__init__.py
basetrainer/criterion/criterion.py
basetrainer/engine/__init__.py
basetrainer/engine/base.py
basetrainer/engine/comm.py
basetrainer/engine/engine.py
basetrainer/engine/launch.py
basetrainer/engine/onnx_engine.py
basetrainer/engine/trainer.py
basetrainer/engine/trt_engine.py
basetrainer/metric/__init__.py
basetrainer/metric/accuracy_recorder.py
basetrainer/metric/eval_tools/__init__.py
basetrainer/metric/eval_tools/acc.py
basetrainer/metric/eval_tools/classification_report.py
basetrainer/metric/eval_tools/eval_utils.py
basetrainer/metric/eval_tools/evaluate.py
basetrainer/metric/eval_tools/iou.py
basetrainer/metric/eval_tools/metrics.py
basetrainer/metric/eval_tools/pandas_tools.py
basetrainer/metric/eval_tools/pr.py
basetrainer/metric/eval_tools/psnr.py
basetrainer/metric/eval_tools/roc.py
basetrainer/metric/eval_tools/verification.py
basetrainer/models/__init__.py
basetrainer/models/build_models.py
basetrainer/optimizer/__init__.py
basetrainer/optimizer/build_optimizer.py
basetrainer/pruning/__init__.py
basetrainer/pruning/nni_pruning.py
basetrainer/pruning/slim_pruning.py
basetrainer/pruning/torch_pruning.py
basetrainer/scheduler/CosineAnnealingLR.py
basetrainer/scheduler/ExponentialLR.py
basetrainer/scheduler/LambdaLR.py
basetrainer/scheduler/MultiStepLR.py
basetrainer/scheduler/MultiStepValue.py
basetrainer/scheduler/WarmUpLR.py
basetrainer/scheduler/__init__.py
basetrainer/scheduler/build_scheduler.py
basetrainer/utils/__init__.py
basetrainer/utils/file_utils.py
basetrainer/utils/image_utils.py
basetrainer/utils/log.py
basetrainer/utils/plot_utils.py
basetrainer/utils/setup_config.py
basetrainer/utils/summary.py
basetrainer/utils/torch_data.py
basetrainer/utils/torch_tools.py
basetrainer/utils/converter/__init__.py
basetrainer/utils/converter/onnx2trt.py
basetrainer/utils/converter/pytorch2onnx.py
test/test_scheduler.py