huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

More realistic training setup. (#210)

Browse Source 
* More realistic training setup.

* Compute the model accuracy.

* Very inefficient backprop for index select.

* More backprop.

* Fix some backprop issues.

* Backprop fix.

* Another broadcasting backprop fix.

* Better backprop for reducing ops.

* Training again.

* Add some gradient tests.

* Get the training to work.
This commit is contained in:
Laurent Mazare
2023-07-20 19:25:41 +02:00
committed by GitHub
parent fa08fb3126
commit 4845d5cc64
6 changed files with 156 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/lib.rs
View File

@ -52,7 +52,7 @@ mod mkl;
pub mod npy;
mod op;
pub mod safetensors;
mod shape;
pub mod shape;
mod storage;
mod strided_index;
mod tensor;