3DTopia / 3DTopia/OpenLRM

训练需要什么样的显卡配置

Open
#36 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.2k
Forks
80
PR merge metrics
No merged PRs in 30d

Description

论文中说训练 我们在128个NVIDIA(40G)A100 GPU上以批次大小1024(每次迭代1024个不同形状)训练LRM 30个周期,大约需要3天时间完成。每个周期包含Objaverse的一份渲染图像数据和MV。
论文里使用的数据集allenai/objaverse 有好几个T

没有这么多a100, 一张a100训练少点数据可以么?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.