首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >使用torch模型/路径文件生成文本到图像

使用torch模型/路径文件生成文本到图像
EN

Stack Overflow用户
提问于 2021-05-31 19:22:31
回答 1查看 74关注 0票数 1

我训练了一个基于https://github.com/aelnouby/Text-to-Image-Synthesis的文本到图像生成模型。现在我有两个路径文件(一个用于生成器,另一个用于鉴别器)。如何使用此路径文件生成图像?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-05-31 19:26:32

你需要在这里传递你的生成器路径文件。trainer.py的self.generator.load_state_dict(torch.load(pre_trained_gen))引用第28行

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67772795

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档