Fix tensor disposal (#6028)

* Upgraded to Sentis 1.3.0-pre.2

* Fixed tensor disposal bug in ModelRunner.

Author: miguelalonsojr

Project/Packages/packages-lock.json

@@ -72,7 +72,7 @@
       "depth": 0,
       "source": "local",
       "dependencies": {
-        "com.unity.sentis": "1.3.0-pre.1",
+        "com.unity.sentis": "1.3.0-pre.2",
         "com.unity.modules.imageconversion": "1.0.0",
         "com.unity.modules.jsonserialize": "1.0.0",
         "com.unity.modules.physics": "1.0.0"
@@ -104,7 +104,7 @@
       "url": "https://packages.unity.com"
     },
     "com.unity.sentis": {
-      "version": "1.3.0-pre.1",
+      "version": "1.3.0-pre.2",
       "depth": 1,
       "source": "registry",
       "dependencies": {

com.unity.ml-agents/CHANGELOG.md

@@ -22,6 +22,7 @@ and this project adheres to
 
 ### Bug Fixes
 #### com.unity.ml-agents / com.unity.ml-agents.extensions (C#)
+- Fixed missing tensor Dispose in ModelRunner (#)
 
 #### ml-agents / ml-agents-envs
 - Bumped numpy version to >=1.21.2,<1.24.0 (#5997)

com.unity.ml-agents/Runtime/Inference/ModelRunner.cs

@@ -150,6 +150,10 @@ public void Dispose()
             if (m_Engine != null)
                 m_Engine.Dispose();
             m_TensorAllocator?.Reset(false);
+            foreach (var (name, tensor) in m_InputsByName)
+            {
+                tensor.Dispose();
+            }
         }
 
         void FetchSentisOutputs(string[] names)