Add prestep and poststep, various fixes around trainer

NVIDIAGameWorks / kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).

Other

1.46k stars 131 forks source link

Add prestep and poststep, various fixes around trainer #77

Closed tovacinni closed 1 year ago

tovacinni commented 1 year ago

Breaking changes: any class that overrides BaseTrainer will have to change a lot of the function signatures to not take in epochs (which were redundant)

Signed-off-by: Towaki Takikawa tovacinni@gmail.com