• Jeremy Reizenstein's avatar
    NeRF training: avoid caching unused visualization data. · f00ef667
    Jeremy Reizenstein authored
    Summary: If we are not visualizing the training with visdom, then there are a couple of outputs of the coarse rendering step which are not small and are returned by the renderer but never used. We don't need to bother transferring them to the CPU.
    
    Reviewed By: nikhilaravi
    
    Differential Revision: D28939958
    
    fbshipit-source-id: 7e0d6681d6524f7fb57b6b20164580006120de80
    f00ef667
train_nerf.py 9.15 KB