yeungchenwa / FontDiffuser

[AAAI2024] FontDiffuser: One-Shot Font Generation via Denoising Diffusion with Multi-Scale Content Aggregation and Style Contrastive Learning
https://yeungchenwa.github.io/fontdiffuser-homepage/
231 stars 21 forks source link

训练出错 #27

Open yang-chenyu104 opened 6 months ago

yang-chenyu104 commented 6 months ago

在训练的时候默认开启了数据并行导致保存DistributedDataParallel,模型model没有unt这个参数导致出错,如何从并行模型提取需要模型参数

yang-chenyu104 commented 6 months ago

加入了model.module就可以,但是训练如何加载你们预训练的模型