search

cesarali / conditional_rate_matching

Flow Matching for Discrete Variables
2 stars 0 forks source link

shape issue when target is noise #4

Closed dfaroughy closed 5 months ago

dfaroughy commented 5 months ago

issue in CRM experiments when using unet for mnist images -> noise (i.e. reverse evolution). The target noise needs to be reshaped to a proper 28 x 28 x 1 image. The issue arises only for unet. When using mlp with flattened images it works fine.

cesarali commented 5 months ago

done