float32
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -37,7 +37,7 @@ def generate3d(model, rgb, ccm, device):
|
|
| 37 |
triplane = torch.cat([color,xyz],dim=1).to(device)
|
| 38 |
# 3D visualize
|
| 39 |
model.eval()
|
| 40 |
-
glctx = dr.RasterizeCudaContext()
|
| 41 |
|
| 42 |
if model.denoising == True:
|
| 43 |
tnew = 20
|
|
|
|
| 37 |
triplane = torch.cat([color,xyz],dim=1).to(device)
|
| 38 |
# 3D visualize
|
| 39 |
model.eval()
|
| 40 |
+
glctx = dr.RasterizeGLContext()#dr.RasterizeCudaContext()
|
| 41 |
|
| 42 |
if model.denoising == True:
|
| 43 |
tnew = 20
|