tiny-cuda-nn
tiny-cuda-nn copied to clipboard
Decouple inference and training.
Hi, I'm playing with mlp_learning_an_image.cu file in the samples directory, it simply trains an MLP and performs inference on the MLP. I want to save the trained model and then load it back (i.e. I want to decouple training from inference). Can you please share some insights on how can I save the model into a file and then load it at inference time? thanks