Free some resources after each step to avoid OOM#75

andrewmoise wants to merge 1 commit intolucidrains:mainlucidrains/denoising-diffusion-pytorch:mainfrom

andrewmoise:mainandrewmoise/denoising-diffusion-pytorch:mainCopy head branch name to clipboard