PRECEPT but in haskell.
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
Build the project with stack
$ stack build
then run it
$ stack run
and observe these sweet succulent plots
... I know it sucks ...
- 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
BSD3