Laurent Mazare af552a5274 Fix the rnn tests for accelerate. (#704)

2023-09-01 13:21:38 +01:00

..

activation.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

batch_norm.rs

VarBuilder cleanup (#627)

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

conv.rs

Dilated convolutions (#657)

2023-08-29 16:12:11 +01:00

embedding.rs

VarBuilder cleanup (#627)

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

func.rs

Add a couple functions required for yolo. (#527)

2023-08-20 17:02:05 +01:00

group_norm.rs

VarBuilder cleanup (#627)

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

init.rs

VarBuilder cleanup (#627)

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

layer_norm.rs

More robust tests (so that they pass on accelerate). (#679)

2023-08-30 18:10:10 +01:00

lib.rs

Add the optimizer trait. (#702)

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

linear.rs

VarBuilder cleanup (#627)

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

loss.rs

Add the cross-entropy loss. (#287)

2023-07-31 14:26:36 +01:00

ops.rs

More robust tests (so that they pass on accelerate). (#679)

2023-08-30 18:10:10 +01:00

optim.rs

Add the optimizer trait. (#702)

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

rnn.rs

Fix the rnn tests for accelerate. (#704)

2023-09-01 13:21:38 +01:00

var_builder.rs

Add some doc to the varbuilder. (#700)

2023-09-01 08:28:35 +01:00

var_map.rs

Move the VarMap to a separate file (#525)

2023-08-20 14:25:07 +01:00