[pytorch] Template 뜯어보기 - [시작]

Motivation

 ipython에서 제공하는 ipynb - notebook을 사용하여 작업을 할 때 많은 장점들이 있다. 바로바로 cell 단위로 결과를 확인해볼 수 있다는 점 그외 에도 user-friendly한 다양한 기능들이 존재한다. 하지만 notebook을 가지고 협업을 하기엔 한계가 있어서 조금 더 파일 관리에 용이한 template을 만들고 싶어졌다.

 

 상기 이유로 pytorch Template으로 유명한 https://github.com/victoresque/pytorch-template 에서 제시하는 Template에서 코드를 하나하나 살펴보고 정리해보고자 한다.

Basic Template

pytorch(template)
│
├── train.py - 훈련목적 main script to start training
├── test.py - evaluation of trained model
│
├── config.json - holds configuration for training
├── parse_config.py - class to handle config file and cli options
│
├── new_project.py - initialize new project with template files
│
├── base/ - abstract base classes
│   ├── base_data_loader.py
│   ├── base_model.py
│   └── base_trainer.py
│
├── data_loader/ - anything about data loading goes here
│   └── data_loaders.py
│
├── data/ - default directory for storing input data
│
├── model/ - models, losses, and metrics
│   ├── model.py
│   ├── metric.py
│   └── loss.py
│
├── saved/
│   ├── models/ - trained models are saved here
│   └── log/ - default logdir for tensorboard and logging output
│
├── trainer/ - trainers
│   └── trainer.py
│
├── logger/ - module for tensorboard visualization and logging
│   ├── visualization.py
│   ├── logger.py
│   └── logger_config.json
│  
└── utils/ - small utility functions
    ├── util.py
    └── ....

---

인스타 주소 🎗

https://www.instagram.com/f.inn_sharp/