Skip to content

lyakaap/pytorch-template

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
Aug 24, 2018
Dec 28, 2018
Mar 30, 2019
Aug 24, 2018
Apr 25, 2019
Aug 24, 2018
Dec 28, 2018
Aug 24, 2018
Aug 24, 2018
Feb 11, 2019

Repository files navigation

pytorch-template

This repository is my PyTorch project template (for Kaggle and research). Currently, this repository supports:

  • some loss functions for semantic segmentation (src/losses.py)
  • mean IoU cuda implementation (src/metrics.py)
  • basic UNet implementation (src/models.py)
  • learning rate scheduler (src/lr_scheduler.py)
  • useful debugging module (src/debug.py)

Enviroments

This repository supports PyTorch >= 1.0. You can setup the enviroment using docker/Dockerfile (Ubuntu 16.04, cuda 9.2).

Run Experiments

Set working directory to "experiments" and run below commands,

# run training and evaluation (with saving checkpoints)
python exp0.py job --devices 0,1

# Run grid-search for better hyperparameter set.
# Parameter space can be set in "exp0.py".
# In below setting, each trial is conducted on a single gpu device 
# and thus whole tuning processes are launched on multiple gpu devices in parallel.
python exp0.py tuning --devices 0,1 --n-gpu 1 --mode 'grid'

About

My PyTorch project template (for Kaggle and research)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published