RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

miccunifi / ladi-vton

[ACM MM 2023] - LaDI-VTON: Latent Diffusion Textual-Inversion Enhanced Virtual Try-On

Other

412 stars 56 forks source link

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0 #27

Closed zainmalik95 closed 1 year ago

zainmalik95 commented 1 year ago

I'm getting this error when trying to run on the DressCode Dataset. can anyone help me on this issue? What I'm doing wrong? here is the command that i used python src/inference.py --dataset dresscode --dresscode_dataroot ./DressCode/DressCode --output_dir ./results --test_order unpaired I've added prints in the dataset.py code where the error was happening and i got this:

ABaldrati commented 1 year ago

Hi @zainmalik95 Do you have this error only on the DressCode dataset?

Alberto