ModuleNotFoundError: No module named 'datasets.dfaust'

[abhi2198]

Hello, I was trying to run generate.py with a custom point-cloud using a pre-trained model but I face this error when I run the code. ModuleNotFoundError: No module named 'datasets.dfaust' this doesn't seem to be a path issue as importing the Dataset class from .shapenet works fine. Any help in the right direction would be appreciated.

[aboulch]

Hello, Thank you for pointing the error. I commented the call to dataset.dfaust. I shall add the DFaust dataset file as soon as I have cleaned the code for release.

[abhi2198]

Great, thanks. There's also another issue when running generate.py, it seems to seek 'checkpoint.pth' file which doesn't exist. Is it something that I have to add or is it a missing file?

[aboulch]

Hello,

The checkpoint.pth should be created while training, saved under results/experiment_name/checkpoint.pth.