Increasing memory - Githubissues

sihyun-yu / PVDM

Official PyTorch implementation of Video Probabilistic Diffusion Models in Projected Latent Space (CVPR 2023).

https://sihyun.me/PVDM

MIT License

287 stars 15 forks source link

Open fhlt opened 1 year ago

fhlt commented 1 year ago

Maybe there is a memory non reclamation issue in the first_stage_train, resulting in gradual memory growth

sihyun-yu commented 11 months ago

Hi,

Thanks for reporting the issue. Maybe there was a minor issue while I am doing a refactoring. Let me fix it if I have time to check this issue.

TousenKaname commented 7 months ago

Hi, I had the same problem. Do you have any solutions?