Ogundepo Odunayo 86e1803191 Add Binary Cross Entropy With Logit Loss to nn crate (#1157) ...

* add bce with logit loss

* add bce with logit loss

* remove imports

* fix tiny bug

* add test documentation and refactor function

* fix test cases and formatting

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

..

activation.rs

Use a silu activation in mistral. (#991)

2023-09-29 07:06:54 +01:00

batch_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

conv.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

embedding.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

func.rs

Make func cloneable. (#1137)

2023-10-20 16:28:50 +01:00

group_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

init.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

layer_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

lib.rs

Add the sequential layer. (#1136)

2023-10-20 16:08:50 +01:00

linear.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +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

Fix the leaky relu. (#898)

2023-09-19 18:17:17 +01:00

optim.rs

Only optimize float tensors. (#1069)

2023-10-10 09:05:41 +01:00

rnn.rs

More general seq forward functions for RNNs. (#1050)

2023-10-07 15:08:01 +01:00

sequential.rs

Make func cloneable. (#1137)

2023-10-20 16:28:50 +01:00

var_builder.rs

feat: add pth varbuilder (#1108)

2023-10-16 16:20:36 +01:00

var_map.rs

Use AsRef<str> for set_one. (#1033)

2023-10-05 06:05:44 +01:00