Tensorflow fixes from PR comments

dotnet · May 12, 2021 · 35ef427 · 35ef427
1 parent ce99388
commit 35ef427
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/Microsoft.ML.Vision/DnnRetrainTransform.cs b/src/Microsoft.ML.Vision/DnnRetrainTransform.cs
@@ -116,7 +116,7 @@ private static DnnRetrainTransformer Create(IHostEnvironment env, ModelLoadConte
                     null, false, addBatchDimensionInput, 1);
             }
 
-            var tempDirPath = Path.GetFullPath(Path.Combine((env as IHostEnvironmentInternal).TempFilePath, nameof(DnnRetrainTransformer) + "_" + Guid.NewGuid()));
+            var tempDirPath = Path.GetFullPath(Path.Combine(((IHostEnvironmentInternal)env).TempFilePath, nameof(DnnRetrainTransformer) + "_" + Guid.NewGuid()));
             CreateFolderWithAclIfNotExists(env, tempDirPath);
             try
             {

diff --git a/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs b/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs
@@ -531,7 +531,7 @@ internal ImageClassificationTrainer(IHostEnvironment env, Options options)
             Host.CheckNonEmpty(options.ScoreColumnName, nameof(options.ScoreColumnName));
             Host.CheckNonEmpty(options.PredictedLabelColumnName, nameof(options.PredictedLabelColumnName));
             tf.compat.v1.disable_eager_execution();
-            _resourcePath = Path.Combine((env as IHostEnvironmentInternal).TempFilePath, "MLNET");
+            _resourcePath = Path.Combine(((IHostEnvironmentInternal)env).TempFilePath, "MLNET");
 
             if (string.IsNullOrEmpty(options.WorkspacePath))
             {
@@ -1320,12 +1320,12 @@ private void AddTransferLearningLayer(string labelColumn,
 
         }
 
-        private static TensorFlowSessionWrapper LoadTensorFlowSessionFromMetaGraph(IHostEnvironment env, Architecture arch)
+        private TensorFlowSessionWrapper LoadTensorFlowSessionFromMetaGraph(IHostEnvironment env, Architecture arch)
         {
             var modelFileName = ModelFileName[arch];
-            var modelFilePath = Path.Combine((env as IHostEnvironmentInternal).TempFilePath, "MLNET", modelFileName);
+            var modelFilePath = Path.Combine(_resourcePath, "MLNET", modelFileName);
             int timeout = 10 * 60 * 1000;
-            DownloadIfNeeded(env, @"meta\" + modelFileName, Path.Combine((env as IHostEnvironmentInternal).TempFilePath, "MLNET"), modelFileName, timeout);
+            DownloadIfNeeded(env, @"meta\" + modelFileName, Path.Combine(_resourcePath, "MLNET"), modelFileName, timeout);
             return new TensorFlowSessionWrapper(GetSession(env, modelFilePath, true), modelFilePath);
         }