/torchtools

Hyper-param parser for torch models

Primary LanguagePythonMIT LicenseMIT

torchtools

Tools for PyTorch

Currenltly consisting of:

  • Hyper-param parser for torch models
  • EarlyStopping