Can i train efficientvit on a single A100 or V100?

mit-han-lab / efficientvit

EfficientViT is a new family of vision models for efficient high-resolution vision.

Apache License 2.0

1.79k stars 164 forks source link

Open sanbuphy opened 11 months ago

sanbuphy commented 11 months ago

Hi , I want to fully retrain the efficientvit model, can you tell me if i can train this model just using the A100 or V100?

carlosuc3m commented 5 months ago

It would be great to have more info about the training!