• Darijan Gudelj's avatar
    load whole dataset in train loop · 37bd280d
    Darijan Gudelj authored
    Summary: Loads the whole dataset and moves it to the device and sends it to for sampling to enable full dataset heterogeneous raysampling.
    
    Reviewed By: bottler
    
    Differential Revision: D39263009
    
    fbshipit-source-id: c527537dfc5f50116849656c9e171e868f6845b1
    37bd280d
training_loop.py 16.3 KB