JeanKaddour / SIN

Causal Effect Inference for Structured Treatments (SIN) (NeurIPS 2021)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Some error during training

yfzhang114 opened this issue · comments

image

image

why return none here?

Thank you for question and interest in our work.

This error is raised within pytorch's DataLoader class (here, by the train_loader instance). So my guess would be that the way you're currently creating the dataset/dataloader does not work properly. Can you post a minimal working example to reproduce this error?

Very interesting work! thanks for your response. I fixed the error by manipulating the input shape and revising the cat_dim function.

image

image

Thank you very much!

I think that these issues arise when using PyG 2 instead of PyG 1.x.
I will try to fix this soon. Thank you for making me aware of this!