new
Browse files
src/models/geometry/render/neural_render.py
CHANGED
|
@@ -74,7 +74,9 @@ class NeuralRender(Renderer):
|
|
| 74 |
def __init__(self, device='cuda', camera_model=None):
|
| 75 |
super(NeuralRender, self).__init__()
|
| 76 |
self.device = device
|
| 77 |
-
self.ctx = dr.RasterizeCudaContext(device=device)
|
|
|
|
|
|
|
| 78 |
self.projection_mtx = None
|
| 79 |
self.camera = camera_model
|
| 80 |
|
|
|
|
| 74 |
def __init__(self, device='cuda', camera_model=None):
|
| 75 |
super(NeuralRender, self).__init__()
|
| 76 |
self.device = device
|
| 77 |
+
# self.ctx = dr.RasterizeCudaContext(device=device)
|
| 78 |
+
self.ctx = dr.RasterizeCudaContext().cuda()
|
| 79 |
+
|
| 80 |
self.projection_mtx = None
|
| 81 |
self.camera = camera_model
|
| 82 |
|