ERROR_TWO

[fuyuyuputao]

when I running train_gaussian.py by cuda in GPU, I think that there is better to have with torch.no_grad(): before for _, dim in enumerate(order): in utils.plot.sample_digits_gaussian. The above advice is to solve the error :runtimeerror: cuda out of memory. Noted that this issue is appearing when you running train_gaussian.py by cuda in GPU.