Update video3d/model_ddp.py

video3d/model_ddp.py

@@ -1154,7 +1154,8 @@ class Unsup3DDDP:
  self.perceptual_loss_use_lin = cfgs.get('perceptual_loss_use_lin', True)
  self.perceptual_loss = lpips.LPIPS(net='vgg', lpips=self.perceptual_loss_use_lin)
 
- self.glctx = dr.RasterizeGLContext()
+ # self.glctx = dr.RasterizeGLContext()
+ self.glctx = dr.RasterizeCudaContext()
  self.render_flow = self.cfgs.get('flow_loss_weight', 0.) > 0.
  self.extra_renders = cfgs.get('extra_renders', [])
  self.renderer_spp = cfgs.get('renderer_spp', 1)