Batch Size more than 1 breaks the code

cyclomon / UNSB

Official Repository of "Unpaired Image-to-Image Translation via Neural Schrödinger Bridge" (ICLR 2024)

MIT License

169 stars 8 forks source link

Closed chichun2002 closed 11 months ago

chichun2002 commented 1 year ago

Batch Size more than 1 breaks the code, tensor sizes mess up

1202kbs commented 11 months ago

Hi, thank you for your interest in our work.

Please refer to our previous answer here.

Best, Beomsu