tencent-ailab / PCDMs

Implementation code:Advancing Pose-Guided Image Synthesis with Progressive Conditional Diffusion Models
Apache License 2.0
150 stars 9 forks source link

tch: - conv_in.weight: found shape torch.Size([320, 4, 3, 3]) in the checkpoint and torch.Size([320, 9, 3, 3]) in the model instantiated #6

Closed I8Robot closed 8 months ago

I8Robot commented 8 months ago

I meet this error: image

muzishen commented 8 months ago

This is a problem with the initial weights, and the code automatically ignores executing down.

I8Robot commented 8 months ago

This is a problem with the initial weights, and the code automatically ignores executing down.

OK,:)