Skip to content

GitLab

Unverified Commit 88358528 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Change leftover AutoNonVariableTypeMode class. (#3897) 

* Change leftover AutoNonVariableTypeMode class.

* Replace with InferenceMode
parent c89cd065
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ios/VisionTestApp/VisionTestApp/ModelRunner.mm
View file @ 88358528
...@@ -33,7 +33,7 @@ static int iter = 20; ...@@ -33,7 +33,7 @@ static int iter = 20;
auto img_tensor = torch::ones({3, 224, 224}, at::ScalarType::Float); auto img_tensor = torch::ones({3, 224, 224}, at::ScalarType::Float);
inputs.push_back(c10::List<at::Tensor>(img_tensor)); inputs.push_back(c10::List<at::Tensor>(img_tensor));
torch::autograd::AutoGradMode guard(false); torch::autograd::AutoGradMode guard(false);
at::AutoNonVariableTypeMode nonVarTypeModeGuard(true); at::InferenceMode nonVarTypeModeGuard(true);
UI_LOG(@"Running warmup runs...", nil); UI_LOG(@"Running warmup runs...", nil);
for (int i = 0; i < warmup; ++i) { for (int i = 0; i < warmup; ++i) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment