huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a8f61e66cc04e2b05caa14a95296ab2b185c3e71
candle/candle-nn/tests
History
Laurent Mazare 42e1cc8062 Add a batch normalization layer (#508) 
* Add BatchNormalization.

* More batch-norm.

* Add some validation of the inputs.

* More validation.
2023-08-18 20:05:56 +01:00
..
batch_norm.rs
Add a batch normalization layer (#508)
2023-08-18 20:05:56 +01:00
group_norm.rs
Add a simple Module trait and implement it for the various nn layers (#500)
2023-08-18 09:38:22 +01:00
layer_norm.rs
Add a simple Module trait and implement it for the various nn layers (#500)
2023-08-18 09:38:22 +01:00
loss.rs
Fix the tests for mkl. (#437)
2023-08-14 08:09:27 +01:00
ops.rs
Add the AdamW optimizer. (#307)
2023-08-02 14:03:49 +01:00
optim.rs
Add a simple Module trait and implement it for the various nn layers (#500)
2023-08-18 09:38:22 +01:00
test_utils.rs
Add the AdamW optimizer. (#307)
2023-08-02 14:03:49 +01:00