nv-tlabs / LION

Latent Point Diffusion Models for 3D Shape Generation
Other
735 stars 57 forks source link

AttributeError: 'Trainer' object has no attribute 'load_vae' #28

Closed fradino closed 1 year ago

fradino commented 1 year ago

When i am training the diffusion prior, it shows: AttributeError: 'Trainer' object has no attribute 'load_vae' script/train_prior.sh: line 20: latent_pts.pvd_mse_loss: command not found @ZENGXH

ZENGXH commented 1 year ago

It looks like the command is break: could you check and make sure you have \ symbol at the end of line 19 in your file, and there should be no space after it.
If it still fail, could you paste your full script/train_prior.sh here?

fradino commented 1 year ago

I fixed "script/train_prior.sh: line 20: latent_pts.pvd_mse_loss: command not found" but AttributeError: 'Trainer' object has no attribute 'load_vae' still exists image The code is here, and I cannot find the function load_vae in this project image

ZENGXH commented 1 year ago

Hi,

I just fix this issue. It should be calling resume function, since it detect a snapshot file under the experiment path. Can you try again?

fradino commented 1 year ago

sure,i will try it right now.