Spaces:
Running
on
Zero
Running
on
Zero
Increase gpu time
Browse files- demo/gs_train.py +1 -1
- demo/mast3r_demo.py +1 -1
demo/gs_train.py
CHANGED
@@ -51,7 +51,7 @@ class ModelParams:
|
|
51 |
data_device: str = "cuda"
|
52 |
eval: bool = False
|
53 |
|
54 |
-
@spaces.GPU(duration=
|
55 |
def train(
|
56 |
data_source_path, iterations, position_lr_init, position_lr_final, position_lr_delay_mult,
|
57 |
position_lr_max_steps, feature_lr, opacity_lr, scaling_lr, rotation_lr,
|
|
|
51 |
data_device: str = "cuda"
|
52 |
eval: bool = False
|
53 |
|
54 |
+
@spaces.GPU(duration=160)
|
55 |
def train(
|
56 |
data_source_path, iterations, position_lr_init, position_lr_final, position_lr_delay_mult,
|
57 |
position_lr_max_steps, feature_lr, opacity_lr, scaling_lr, rotation_lr,
|
demo/mast3r_demo.py
CHANGED
@@ -173,7 +173,7 @@ def save_colmap_scene(scene, save_dir, min_conf_thr=2, clean_depth=False, mask_i
|
|
173 |
save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
|
174 |
return save_path
|
175 |
|
176 |
-
@spaces.GPU(duration=
|
177 |
def get_reconstructed_scene(snapshot, current_scene_state,
|
178 |
min_conf_thr, matching_conf_thr,
|
179 |
as_pointcloud, cam_size, shared_intrinsics, clean_depth, filelist, req: gradio.Request, **kw):
|
|
|
173 |
save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
|
174 |
return save_path
|
175 |
|
176 |
+
@spaces.GPU(duration=160)
|
177 |
def get_reconstructed_scene(snapshot, current_scene_state,
|
178 |
min_conf_thr, matching_conf_thr,
|
179 |
as_pointcloud, cam_size, shared_intrinsics, clean_depth, filelist, req: gradio.Request, **kw):
|