search

HuiZhang0812 / DiffusionAD

148 stars 16 forks source link

训练与推理时间 #49

Closed yuki1ssad closed 3 months ago

yuki1ssad commented 4 months ago

您好，请问可否告知硬件配置及相应的训练时长与推理速度？谢谢！

Super-mylife commented 4 months ago

您好，请问可否告知硬件配置及相应的训练时长与推理速度？谢谢！

您好您有尝试嘛 我初步训练了一下单卡24gb下只能跑最大3batch 跑完可能需要半个月...

HuiZhang0812 commented 4 months ago

We trained on A100 80g, and the average training time was about 16 hours.

caihaocong commented 3 months ago

可以修改参数，缩小模型，例如: base_channel，实际不需要那么大的模型也能有不错的效果的