Output images with gradient during inference

[lelechen63]

Hi there,

I try to output the image with the gradient. However, I found that if I use your default testing code, it will call whole_grad_forward (

CIPS-3D/exp/dev/nerf_inr/models/generator_nerf_inr_v16.py

Line 1395 in aee4025

with torch.no_grad():

), and will remove the gradient. If I comment out the torch.no_grad(), it would be out of memory. Is there a way to output the image with gradient? Thanks

[lelechen63]

Solved.