stash

Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com>

Author: borisfom

vista3d/modeling/vista3d.py

@@ -310,15 +310,14 @@ def forward(
             )
             torch.cuda.synchronize()
             print(f"Encoder Time: {time.time() - time0}, shape : {input_images.shape}, point: {point_coords is not None}")
-            if False: 
+            if self.engine is None: 
                 # breakpoint()
                 torch.onnx.export(self.image_encoder,
                                   (input_images,),
                                   "Encoder.onnx",
                                   verbose=False,
                                   opset_version=18
                                   )
-                self.engine = True
 
         input_images = None
         time1 = time.time()