我训练了一个基于https://github.com/aelnouby/Text-to-Image-Synthesis的文本到图像生成模型。现在我有两个路径文件(一个用于生成器,另一个用于鉴别器)。如何使用此路径文件生成图像?
发布于 2021-05-31 19:26:32
你需要在这里传递你的生成器路径文件。trainer.py的self.generator.load_state_dict(torch.load(pre_trained_gen))引用第28行
https://stackoverflow.com/questions/67772795
复制相似问题