Skip to content

AugustUnderground/prehsept

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

prehsept

PRECEPT but in haskell.

Setup

Requires libtorch symlinked into the root of this repository according to extra-include-dirs and extra-lib-dirs in stack.yaml.

The setenv script can be sourced to export the correct LD_LIBRARY_PATH. Additionally it exports the DEVICE variable, supposedly enabling GPU support in Hasktorch (I'm not sure how/if it works).

$ source setenv

Usage

Build the project with stack

$ stack build

then run it

$ stack run

and observe these sweet succulent plots

... I know it sucks ...

TODO

  • Fix validation loop
  • Add argument parser for training data path etc.
  • Add Tests.
  • Weighted data sampling
  • Better weight init (xavier maybe?)
  • Try data loaders.
  • Move training loop to Lib.
  • Plots
  • Serving

License

BSD3

Thanks

Thanks to the hastorch project and the stack sekelton!