Hello, I’m using ViT from your packages, in particular I’m using facebook/deit-base-patch16-384, when I try and test it on CPU all the workflow goes well but when I test it on GPU I receive this error: Using GPU Trainable param: 105521444 Traceback (most recent call last): File “D:\WorkingDirec…

ViT problem with GPU usage require image to be numpy

marcomameli01 June 24, 2022, 9:40pm 4

I can share my github code, but privately for the moment and you can reproduce the environment with the data. If you agree please let me know your mail.

Topic		Replies	Views
ViT Model increasing CPU RAM when moving to GPU 🤗Transformers	0	226	August 12, 2022
HuggingFace ViT 10x Slower than Native Tensorflow (Not Fully Using GPU?) 🤗Transformers	0	347	July 16, 2022
InvalidArgumentError with vit-base-patch16-224 model? Models	7	2011	April 20, 2024
Can't Load ViT Model for Fine Tuning 🤗Transformers	2	1510	August 11, 2022
"too many values to unpack (expected 4)" but pixel_values dimension is correct 🤗Transformers	2	425	February 14, 2024

ViT problem with GPU usage require image to be numpy

Related topics