Skip to content

Commit

Permalink
Update readme on notebooks
Browse files Browse the repository at this point in the history
  • Loading branch information
braun-steven committed Nov 5, 2023
1 parent 46daf5d commit 558379c
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,14 @@ This repository contains code for my personal, simplistic, EinsumNetworks implem

For a speed benchmark comparison against the official EinsumNetworks implementation, check out [benchmark.md](./benchmark/benchmark.md) (short: simple-einet is faster in all dimensions except the input-channel size but scales similar to EinsumNetworks).

# PyTorch Lightning Training
## Notebooks

The `notebooks` folder contains some Jupyter notebooks that demonstrate the usage of this library.

- [Training an Einet on MNIST](./notebooks/mnist.ipynb)
- [Training an Einet on synthetic multivariate Normal data](./notebooks/multivariate_normal.ipynb)

## PyTorch Lightning Training

The `main_pl.py` script offers PyTorch-Lightning based training for discriminative and generative Einets.

Expand Down

0 comments on commit 558379c

Please sign in to comment.