ostapagon commited on
Commit
1fe4c34
·
1 Parent(s): d0be95c

Increase gpu time

Browse files
Files changed (3) hide show
  1. demo/gs_demo.py +1 -1
  2. demo/gs_train.py +1 -1
  3. 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=9000, minimum=1, maximum=15000, step=5)
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=30)
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=30)
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):