Hi, I found that both torchscript and torch::nn::Module model file can be loaded with torch::load(model, "model_file.pt");. It isn’t nesscery to save torchscript model in cpp while you load torchscript model to nn::Module and train it.