comment torch.jit

layumi · May 11, 2023 · a4a33f1 · a4a33f1
1 parent 36cac9a
commit a4a33f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.test_MSMT.py.swp b/.test_MSMT.py.swp
diff --git a/test.py b/test.py
@@ -303,8 +303,8 @@ def get_id(img_path):
 # To do so, we can call `.trace` on the reparamtrized module with dummy inputs
 # expected by the module.
 # Comment out this following line if you do not want to trace.
-dummy_forward_input = torch.rand(opt.batchsize, 3, h, w).cuda()
-model = torch.jit.trace(model, dummy_forward_input)
+#dummy_forward_input = torch.rand(opt.batchsize, 3, h, w).cuda()
+#model = torch.jit.trace(model, dummy_forward_input)
 
 print(model)
 # Extract feature