├── .flake8 ├── .github └── PULL_REQUEST_TEMPLATE.md ├── .gitignore ├── .gitmodules ├── .travis.yml ├── README.md ├── datasets └── .gitkeep ├── doc ├── PytorchTemplate-initialDesgin.png └── UseThisTemplate.png ├── environment.yaml ├── libs └── .gitkeep ├── pretrained_model └── .gitkeep └── src ├── LICENSE ├── configs ├── arch │ └── mnist.json ├── data │ └── mnist.json ├── loss │ └── cross_entropy.json ├── template_GAN_train_config.json ├── template_eval_config.json ├── template_test_config.json ├── template_train_config.json └── train │ └── finetune_first_conv.json ├── data_loader ├── __init__.py ├── base_data_loader.py ├── data_loaders.py ├── mnist.py └── mnist_result.py ├── main.py ├── model ├── __init__.py ├── base_model.py ├── loss.py ├── metric.py ├── model.py └── networks.py ├── pipeline ├── __init__.py ├── base_pipeline.py ├── evaluation_pipeline.py ├── testing_pipeline.py └── training_pipeline.py ├── scripts ├── clean_saved.py └── make_dataset_list.py ├── utils ├── __init__.py ├── global_config.py ├── logging_config.py ├── util.py └── visualization.py └── worker ├── evaluator.py ├── tester.py ├── trainer.py ├── training_worker.py ├── validator.py └── worker_template.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/.gitmodules -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/.travis.yml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/README.md -------------------------------------------------------------------------------- /datasets/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /doc/PytorchTemplate-initialDesgin.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/doc/PytorchTemplate-initialDesgin.png -------------------------------------------------------------------------------- /doc/UseThisTemplate.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/doc/UseThisTemplate.png -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/environment.yaml -------------------------------------------------------------------------------- /libs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained_model/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/LICENSE -------------------------------------------------------------------------------- /src/configs/arch/mnist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/arch/mnist.json -------------------------------------------------------------------------------- /src/configs/data/mnist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/data/mnist.json -------------------------------------------------------------------------------- /src/configs/loss/cross_entropy.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/loss/cross_entropy.json -------------------------------------------------------------------------------- /src/configs/template_GAN_train_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/template_GAN_train_config.json -------------------------------------------------------------------------------- /src/configs/template_eval_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/template_eval_config.json -------------------------------------------------------------------------------- /src/configs/template_test_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/template_test_config.json -------------------------------------------------------------------------------- /src/configs/template_train_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/template_train_config.json -------------------------------------------------------------------------------- /src/configs/train/finetune_first_conv.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/configs/train/finetune_first_conv.json -------------------------------------------------------------------------------- /src/data_loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data_loader/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/data_loader/base_data_loader.py -------------------------------------------------------------------------------- /src/data_loader/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/data_loader/data_loaders.py -------------------------------------------------------------------------------- /src/data_loader/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/data_loader/mnist.py -------------------------------------------------------------------------------- /src/data_loader/mnist_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/data_loader/mnist_result.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/main.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/model/base_model.py -------------------------------------------------------------------------------- /src/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/model/loss.py -------------------------------------------------------------------------------- /src/model/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/model/metric.py -------------------------------------------------------------------------------- /src/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/model/model.py -------------------------------------------------------------------------------- /src/model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/model/networks.py -------------------------------------------------------------------------------- /src/pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/pipeline/__init__.py -------------------------------------------------------------------------------- /src/pipeline/base_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/pipeline/base_pipeline.py -------------------------------------------------------------------------------- /src/pipeline/evaluation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/pipeline/evaluation_pipeline.py -------------------------------------------------------------------------------- /src/pipeline/testing_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/pipeline/testing_pipeline.py -------------------------------------------------------------------------------- /src/pipeline/training_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/pipeline/training_pipeline.py -------------------------------------------------------------------------------- /src/scripts/clean_saved.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/scripts/clean_saved.py -------------------------------------------------------------------------------- /src/scripts/make_dataset_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/scripts/make_dataset_list.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/global_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/utils/global_config.py -------------------------------------------------------------------------------- /src/utils/logging_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/utils/logging_config.py -------------------------------------------------------------------------------- /src/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/utils/util.py -------------------------------------------------------------------------------- /src/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/utils/visualization.py -------------------------------------------------------------------------------- /src/worker/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/evaluator.py -------------------------------------------------------------------------------- /src/worker/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/tester.py -------------------------------------------------------------------------------- /src/worker/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/trainer.py -------------------------------------------------------------------------------- /src/worker/training_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/training_worker.py -------------------------------------------------------------------------------- /src/worker/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/validator.py -------------------------------------------------------------------------------- /src/worker/worker_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amjltc295/pytorch-golden-template/HEAD/src/worker/worker_template.py --------------------------------------------------------------------------------