训练显存要求和训练时长

[xiansen11]

想问一下训练和推理要求显卡最低显存是多少呢？要训练多久呢？比pixart-a的训练时间长还是短呢？

[HongkLin]

论文中200k iteration的模型训练用的是4张24G 4090，大概两天，推理一张就够。和pixart-alpha本身的训练不具备可比性，因为训练目标，设备，batch size等设置都不一样，我们是在pixart-alpha基础上fine-tune的。

[xiansen11]

论文中200k iteration的模型训练用的是4张24G 4090，大概两天，推理一张就够。和pixart-alpha本身的训练不具备可比性，因为训练目标，设备，batch size等设置都不一样，我们是在pixart-alpha基础上fine-tune的。

想问一下，用的是这个预训练的权重文件吗？

[HongkLin]

是的，不过用的是diffusers版本的

[xiansen11]

是的，不过用的是diffusers版本的

你好，想问问，是要把这些都下载，然后放到./model_weights目录下吗？有./model_weights的文件组织结构可以看一下吗？