What resources to train from scratch

3DTopia / OpenLRM

An open-source impl. of Large Reconstruction Models

Apache License 2.0

947 stars 52 forks source link

Open reynoldscem opened 6 months ago

reynoldscem commented 6 months ago

Hi,

Could you please provide some numbers around which GPUs, how many of them, and how many hours roughly were required to train these models?

ZexinHe commented 5 months ago

Hi,

Plz refer to this issue https://github.com/3DTopia/OpenLRM/issues/2#issuecomment-1882590904. There isn't much differences for V1.1 models.

Plz also note that you can also use less resources, e.g. 8 A100 GPUs, by configuring the gradient accumulation steps in the config file.