Laurent Mazare b5c283e86f Add the prelu layer. (#1402)

2023-12-03 16:06:09 +00:00

..

activation.rs

Add the prelu layer. (#1402)

2023-12-03 16:06:09 +00:00

batch_norm.rs

Add fuse-conv-bn method for Conv2d (#1196)

2023-10-27 15:56:50 +01:00

conv.rs

Transposed conv1d in candle-nn. (#1252)

2023-11-03 11:18:25 +01:00

embedding.rs

Add clone to various nn layers. (#910)

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

func.rs

Allow for different behavior between training and eval (#1213)

2023-10-29 07:53:09 +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 weight and bias functions to LayerNorm (#1306)

2023-11-09 16:09:01 +01:00

lib.rs

Add the prelu layer. (#1402)

2023-12-03 16:06:09 +00:00

linear.rs

Add the prelu layer. (#1402)

2023-12-03 16:06:09 +00: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 the swiglu activation from the chatglm PR. (#1246)

2023-11-02 20:01:34 +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