Output images with gradient during inference - Githubissues

PeterouZh / CIPS-3D

3D-aware GANs based on NeRF (arXiv).

MIT License

610 stars 60 forks source link

Output images with gradient during inference #25

Closed lelechen63 closed 2 years ago

lelechen63 commented 2 years ago

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 (https://github.com/PeterouZh/CIPS-3D/blob/aee40251a02c34e58d3002bcb845151c41b538f0/exp/dev/nerf_inr/models/generator_nerf_inr_v16.py#L1395), 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 commented 2 years ago

Solved.