nkoppel 4290b81244 [Breaking] Add training to batchnorm with exponential moving average (#1504) ...

* Add training to batchnorm with exponential moving average

* Add more checks to batch norm

* Resolve some review comments

* Add with_momentum varients of `new` methods

* Add check for range of momentum variable; update batch norm test

* Run cargo fmt

* Add back num_features parameter

* Format; tiny simplification

2023-12-30 16:42:08 +01:00

..

batch_norm.rs

[Breaking] Add training to batchnorm with exponential moving average (#1504)

2023-12-30 16:42:08 +01:00

group_norm.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

layer_norm.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

loss.rs

Add Binary Cross Entropy With Logit Loss to nn crate (#1157)

2023-10-23 17:12:44 +01:00

ops.rs

Add a custom softmax implementation. (#744)

2023-09-05 14:20:23 +01:00

optim.rs

Add the optimizer trait. (#702)

2023-09-01 12:55:39 +01:00

rnn.rs

Add a GRU layer. (#688)

2023-08-31 08:43:10 +01:00