Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
bc4bba8
1
Parent(s):
e6feb9e
Tweak to demo
Browse files
demo.py
CHANGED
|
@@ -47,9 +47,9 @@ def get_reconstructed_scene(outdir, model, device, silent, image_size, ios_mode,
|
|
| 47 |
if __name__ == '__main__':
|
| 48 |
|
| 49 |
image_size = 512
|
| 50 |
-
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 51 |
silent = False
|
| 52 |
ios_mode = True
|
|
|
|
| 53 |
|
| 54 |
model_name = "brandonsmart/splatt3r_v1.0"
|
| 55 |
filename = "epoch=19-step=1200.ckpt"
|
|
|
|
| 47 |
if __name__ == '__main__':
|
| 48 |
|
| 49 |
image_size = 512
|
|
|
|
| 50 |
silent = False
|
| 51 |
ios_mode = True
|
| 52 |
+
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 53 |
|
| 54 |
model_name = "brandonsmart/splatt3r_v1.0"
|
| 55 |
filename = "epoch=19-step=1200.ckpt"
|