search

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
272 stars 35 forks source link

About the shape #10

Closed xuecheng990531 closed 1 year ago

xuecheng990531 commented 1 year ago

assert x_t.shape == eps.shape AssertionError

Do I need to change the learn_sigma=True?

JuliaWolleb commented 1 year ago

Well, in what line does this happen? what is x_t.shape and eps.shape?

xuecheng990531 commented 1 year ago

I solved this problem yesterday and I think that when determining the x_t shape and eps shape, I only need to determine whether their channel numbers are the same.

sherrlockk commented 3 months ago

How did you solve this problem？I'm having the same issue

xuecheng990531 commented 2 months ago

时间太长了，我都忘记了哈哈