torch_snippet

Personal templates to train model on torch

Features:

1. auto argparser
2. a logger which prints info in console and saves info into files
3. auto set seed

How to use:

1. just change the project_path in main.py as your project path;
2. to print logs, use logger.info(msg)