Spaces:
Running
on
Zero
Running
on
Zero
Increase gpu time
Browse files- demo/gs_demo.py +1 -1
- demo/gs_train.py +1 -1
- demo/mast3r_demo.py +1 -1
demo/gs_demo.py
CHANGED
@@ -88,7 +88,7 @@ def gs_demo_tab():
|
|
88 |
densify_from_iter = gr.Number(label="Densify From Iter", value=500)
|
89 |
densify_until_iter = gr.Number(label="Densify Until Iter", value=15000)
|
90 |
densify_grad_threshold = gr.Number(label="Densify Grad Threshold", value=0.0002)
|
91 |
-
iterations = gr.Slider(label="Iterations", value=
|
92 |
|
93 |
start_button = gr.Button("Start Training")
|
94 |
|
|
|
88 |
densify_from_iter = gr.Number(label="Densify From Iter", value=500)
|
89 |
densify_until_iter = gr.Number(label="Densify Until Iter", value=15000)
|
90 |
densify_grad_threshold = gr.Number(label="Densify Grad Threshold", value=0.0002)
|
91 |
+
iterations = gr.Slider(label="Iterations", value=8000, minimum=1, maximum=15000, step=5)
|
92 |
|
93 |
start_button = gr.Button("Start Training")
|
94 |
|
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=170)
|
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, **kw):
|
|
|
173 |
save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
|
174 |
return save_path
|
175 |
|
176 |
+
@spaces.GPU(duration=170)
|
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, **kw):
|