I am doing an android model training with onnxruntime training. Now I need to update checkpoint (the model parameters) every round by recreating the OrtTrainingSession object. But the previous OrtTrainingSession object did not free the resource. The running memory of the VM increases rapidly.What should I do
Execution per round
ortTrainingSession.close();
this.ortTrainingSession = this.ortEnv.createTrainingSession(this.newCheckpointPath,
this.trainingModelPath, this.evalModelPath, this.optimizerModelPath);
user24769812 is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.