Change leftover AutoNonVariableTypeMode class. (#3897)

* Change leftover AutoNonVariableTypeMode class. * Replace with InferenceMode

Change leftover AutoNonVariableTypeMode class. (#3897)
* Change leftover AutoNonVariableTypeMode class. * Replace with InferenceMode
88358528 · Vasilis Vryniotis · GitHub · c89cd065 · 88358528
Unverified Commit 88358528 authored May 24, 2021 by Vasilis Vryniotis Committed by GitHub May 24, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ios/VisionTestApp/VisionTestApp/ModelRunner.mm ios/VisionTestApp/VisionTestApp/ModelRunner.mm +1 -1

No files found.
--- a/ios/VisionTestApp/VisionTestApp/ModelRunner.mm
+++ b/ios/VisionTestApp/VisionTestApp/ModelRunner.mm
@@ -33,7 +33,7 @@ static int iter = 20;
  auto img_tensor = torch::ones({3, 224, 224}, at::ScalarType::Float);
  inputs.push_back(c10::List<at::Tensor>(img_tensor));
  torch::autograd::AutoGradMode guard(false);
-  at::AutoNonVariableTypeMode nonVarTypeModeGuard(true);
+  at::InferenceMode nonVarTypeModeGuard(true);
  
  UI_LOG(@"Running warmup runs...", nil);
  for (int i = 0; i < warmup; ++i) {