huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f25173d68b26d8915131896fc38c12167a45a3df
candle/candle-core
History
Ivor Wanders f25173d68b Fix for backprop in ConvTranspose2D with stride of 2 (#2337) 
* Add gradient test for conv_transpose2d with stride of 2.

* Swap dilation and stride in ConvTranspose2D backpropagation.

Without this, a shape mismatch occurs with a stride of 2 and dilation of 1.

* Add further tests of the ConvTranspose2D gradient.

Values calculated with torch, minor numerical errors adjusted and commented.
2024-07-17 19:22:23 +02:00
..
benches
Metal Unary: Add benchmarks and process kernels in a tile based fashion (#2056)
2024-04-21 00:10:33 +02:00
examples
Add a basic metal example with capture (#2324)
2024-07-09 12:38:11 +02:00
src
Fix for backprop in ConvTranspose2D with stride of 2 (#2337)
2024-07-17 19:22:23 +02:00
tests
Fix for backprop in ConvTranspose2D with stride of 2 (#2337)
2024-07-17 19:22:23 +02:00
Cargo.toml
Add a basic metal example with capture (#2324)
2024-07-09 12:38:11 +02:00
LICENSE
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
README.md
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00

README.md

candle

Minimalist ML framework for Rust