huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the candle-datasets crate (#322)

Browse Source 
* Move the vision datasets to a separate crate.

* Move the batcher bits.

* Update the readme.

* Move the tiny-stories bits.

---------

Co-authored-by: Jane Doe <jane.doe@example.org>
This commit is contained in:
Laurent Mazare
2023-08-05 08:56:50 +01:00
committed by GitHub
parent f7b2a0391d
commit 620f83cf66
15 changed files with 163 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,7 +2,6 @@
// error type if needed or add some specialized cases on the candle-core side.
pub mod activation;
pub mod conv;
pub mod dataset;
pub mod embedding;
pub mod init;
pub mod layer_norm;
@ -11,7 +10,6 @@ pub mod loss;
pub mod ops;
pub mod optim;
pub mod var_builder;
pub mod vision;
pub use activation::Activation;
pub use conv::{Conv1d, Conv1dConfig};