/TorchUtils

Implementations of commonly used utilities in Pytorch, to be used in various other projects that require them.

Primary LanguagePython

Torch-Utils

Implementations of commonly used utilities in Pytorch, to be used in my various other projects that require them.

Currently Added:

  • Early Stopping. Boolean flag which denotes whether training should continue.
  • Stats tracker which keeps track of loss metrics and stores the best model throughout training
  • OneHot Encoder class.

To add:

  • Add checkpointing
  • Add tensorboard utils (and by extension, figure out how to do basic tensorboard with torch)