3 Commits

Author	SHA1	Message	Date
Laurent Mazare	cb069d6063	Add the permute op (similar to pytorch). (#504) ... * Add the permute op (similar to pytorch). * Add the backprop for dimension permutation.	2023-08-18 16:30:53 +01:00
Laurent Mazare	4f1541526c	dinov2 - read images from disk and compute the class probabilities (#503) ... * Load the image from disk and convert it to a tensor. * Tweak the function name.	2023-08-18 15:50:33 +01:00
Laurent Mazare	95462c6a2e	Add a vision transformer example (dino-v2). (#502) ... * Add a vision transformer example (dino-v2). * Add some documentation + test. * CI fix. * Another fix (still unable to replicate the errors locally :( )	2023-08-18 11:58:06 +01:00