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

Add a vision transformer example (dino-v2). (#502)

Browse Source 
* Add a vision transformer example (dino-v2).

* Add some documentation + test.

* CI fix.

* Another fix (still unable to replicate the errors locally :( )
This commit is contained in:
Laurent Mazare
2023-08-18 11:58:06 +01:00
committed by GitHub
parent b9661a1c25
commit 95462c6a2e
4 changed files with 351 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-book/src/guide/hello_world.md
View File

@ -147,7 +147,7 @@ And rewrite our examples using it
# extern crate candle_core;
# extern crate candle_nn;
use candle_core::{DType, Device, Result, Tensor};
use candle_nn::Linear;
use candle_nn::{Linear, Module};
struct Model {
first: Linear,