GPU Requirement to use vision transformer

google-research / vision_transformer

Apache License 2.0

10.18k stars 1.27k forks source link

Open nowhash opened 1 year ago

nowhash commented 1 year ago

Is it possible to use vision transformer properly on a 12 gb GPU machine?

SRDdev commented 7 months ago

Yes, try reducing your batch_size or reducing the num_encoder_layers