Laurent Mazare 607ffb9f1e Retrieve more information from PyTorch checkpoints. (#515) ...

* Retrieve more information from PyTorch checkpoints.

* Add enough support to load dino-v2 backbone weights.

2023-08-19 15:05:34 +01:00

..

basics.rs

Add some conv1d test + bugfix using padding. (#349)

2023-08-08 20:50:20 +01:00

cpu_benchmarks.rs

Tensor -> QTensor conversion (#496)

2023-08-18 08:19:20 +01:00

cuda_basics.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

cuda_sum_benchmark.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00

tensor-tools.rs

Retrieve more information from PyTorch checkpoints. (#515)

2023-08-19 15:05:34 +01:00