zachcp b4deb5c5a9 Provide a method to allow PTH files with state maps to be loaded. (#2639) ...

* Provide a method to allow PTH files iwth state maps to be loaded.

* add a line to the doc

* String-. &str

2024-11-26 22:52:53 +01:00

..

activation.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

batch_norm.rs

Detach the tensors on batch-norm eval. (#1702)

2024-02-13 14:26:32 +01:00

conv.rs

Bugfix for applying the bias in conv1d-transpose. (#1732)

2024-02-18 22:51:20 +01:00

embedding.rs

Add clone to various nn layers. (#910)

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

encoding.rs

Fix clippy lints for 1.76. (#1682)

2024-02-08 16:48:47 +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

kv_cache.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

layer_norm.rs

Enable the new layer-norm. (#2213)

2024-05-24 16:48:21 +02:00

lib.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

linear.rs

Support for attention bias in gemma + refactor things a bit. (#1744)

2024-02-22 09:35:28 +01:00

loss.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

ops.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

optim.rs

Expose AdamW parameters (#1449)

2023-12-16 18:41:56 -06:00

rnn.rs

Make the RNN configs accessible from the models. (#2541)

2024-10-04 14:22:23 +02:00

rotary_emb.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

sequential.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

var_builder.rs

Provide a method to allow PTH files with state maps to be loaded. (#2639)

2024-11-26 22:52:53 +01:00

var_map.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00