opensora1.1训练
收藏回复举报
opensora1.1训练
t('forum.solved') 已解决
发表于2024-08-22 14:37:36
0 查看

opensora1.1训练问题

设备:A800T A2

参考链接:https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/PyTorch/built-in/mlm/OpenSora1.1

问题1:我们测试发现这边只有一阶段的stage1的训练,没有后面stage2/stage3的训练。这个后面会优化吗?

cke_3272.png

https://github.com/hpcaitech/Open-Sora/blob/main/docs/zh_CN/report_v2.md

cke_4112.png

问题2:

单机单卡推理,我发现训练过后的文件 或者下载的文件都没有这个model.pth

cke_8074.png

训练后生成文件

cke_27537.png

下载的

cke_32794.png

我要发帖子