Skip to content

enijkamp/pygrid

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pygrid

Minimalistic multi-gpu / cpu grid search for pytorch.

requires:

  • python >= 3.6
  • pytorch

features:

  • csv: params, status, results are contained in a csv file
  • multi-processing: distributes isolated pytorch processes among gpus / cpus
  • timestamps: isolates jobs in timestamped output directories
  • logging: isolates logging facility per job

setup:

  1. fill in TODOs with your training code and list of parameters
  2. set list of gpu / cpu device ids
  3. run

About

Hyperparameter search for pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages