This change set introduces the WispModule interface. All important base building blocks should subclass this interface: acceleration structures, grids (and their internal components like feature volume for triplane), decoders, position embedders, neural fields, bottom level renderers and so on.
The WispModule interface is fairly thin, it assumes a module:
Is a torch nn.Module
Knows how to return a dict of useful properties, this is useful for logging / gui purposes.
Optional: Knows how to return a human readable name (default is to take the class name, where this makes sense)
This change set introduces the
WispModuleinterface. All important base building blocks should subclass this interface: acceleration structures, grids (and their internal components like feature volume for triplane), decoders, position embedders, neural fields, bottom level renderers and so on.The
WispModuleinterface is fairly thin, it assumes a module:nn.ModuleSigned-off-by: operel operel@nvidia.com