Hi,
When I run : !python scripts/pipeline.py --config /content/drive/MyDrive/tab-ddpm/data/industry/config.toml --train --sample
The error occured:
mlp
File "/content/drive/MyDrive/tab-ddpm/scripts/tab_ddpm/modules.py", line 470, in forward
emb += F.silu(self.label_emb(y))
RuntimeError: The size of tensor a (256) must match the size of tensor b (39) at non-singleton dimension 1
and I print out the emb size which is torch.Size([256, 128]); and label_emb(y) size is [256,39,128], where y shape is [256,39];
I tried to replace emb += F.silu(self.label_emb(y)) with emb += F.silu(self.label_emb(y.sequeeze())) as ResNetDiffusion did in module.py, but it does not work, still print out the same error message.
Hi, When I run : !python scripts/pipeline.py --config /content/drive/MyDrive/tab-ddpm/data/industry/config.toml --train --sample
The error occured: mlp File "/content/drive/MyDrive/tab-ddpm/scripts/tab_ddpm/modules.py", line 470, in forward emb += F.silu(self.label_emb(y)) RuntimeError: The size of tensor a (256) must match the size of tensor b (39) at non-singleton dimension 1
and I print out the emb size which is torch.Size([256, 128]); and label_emb(y) size is [256,39,128], where y shape is [256,39];
I tried to replace emb += F.silu(self.label_emb(y)) with emb += F.silu(self.label_emb(y.sequeeze())) as ResNetDiffusion did in module.py, but it does not work, still print out the same error message.