Update test_latency.py

huawei-noah · Jun 6, 2023 · b6b9bf7 · b6b9bf7
1 parent 0be0372
commit b6b9bf7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test_latency.py b/test_latency.py
@@ -20,9 +20,8 @@
     size = 224
     data_loader_val = create_loader(dataset_val, input_size=size, batch_size=1, is_training=False, use_prefetcher=False)
 
-    net = vanillanet_5().cuda()
+    net = vanillanet_5(deploy=True).cuda()
     net.eval()
-    net.switch_to_deploy()
     print(net)
     for img, target in data_loader_val:
         img = img.cuda()