NetSmith currently exports clean PyTorch modules.
I can expand the export options — what would you prefer?
Standard nn.Module
Lightning-style modules
Scriptable modules
Custom template support
0
Add a comment
Replies