Skip to content

Commit

Permalink
fix: accidental deletion
Browse files Browse the repository at this point in the history
  • Loading branch information
avik-pal authored Sep 17, 2024
1 parent 99c3370 commit 1a9f7e1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/src/examples/hamiltonian_nn.md
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ dataloader = ncycle(
NEPOCHS)
```

### Training the
### Training the HamiltonianNN

We parameterize the with a small MultiLayered Perceptron. HNNs are trained by optimizing the gradients of the Neural Network. Zygote currently doesn't support nesting itself, so we will be using ForwardDiff in the training loop to compute the gradients of the HNN Layer for Optimization.

Expand Down

0 comments on commit 1a9f7e1

Please sign in to comment.