ostapagon commited on
Commit
be4fd4f
·
1 Parent(s): 9c1bae1

Change quota, fix paths

Browse files
Files changed (2) hide show
  1. demo/gs_train.py +1 -1
  2. demo/mast3r_demo.py +1 -1
demo/gs_train.py CHANGED
@@ -60,7 +60,7 @@ class TrainingArgs:
60
  checkpoint_iterations: list[int] = field(default_factory=lambda: [7_000, 15_000, 30_000])
61
  start_checkpoint: str = None
62
 
63
- @spaces.GPU(duration=180)
64
  def train(
65
  data_source_path, sh_degree, model_path, images, resolution, white_background, data_device, eval,
66
  convert_SHs_python, compute_cov3D_python, debug,
 
60
  checkpoint_iterations: list[int] = field(default_factory=lambda: [7_000, 15_000, 30_000])
61
  start_checkpoint: str = None
62
 
63
+ @spaces.GPU(duration=20)
64
  def train(
65
  data_source_path, sh_degree, model_path, images, resolution, white_background, data_device, eval,
66
  convert_SHs_python, compute_cov3D_python, debug,
demo/mast3r_demo.py CHANGED
@@ -174,7 +174,7 @@ def save_colmap_scene(scene, save_dir, min_conf_thr=2, clean_depth=False):
174
  save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
175
  return save_path
176
 
177
- @spaces.GPU(duration=180)
178
  def get_reconstructed_scene(outdir, model, device, silent, image_size, current_scene_state,
179
  filelist, optim_level, lr1, niter1, lr2, niter2, min_conf_thr, matching_conf_thr,
180
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, scenegraph_type, winsize,
 
174
  save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
175
  return save_path
176
 
177
+ @spaces.GPU(duration=20)
178
  def get_reconstructed_scene(outdir, model, device, silent, image_size, current_scene_state,
179
  filelist, optim_level, lr1, niter1, lr2, niter2, min_conf_thr, matching_conf_thr,
180
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, scenegraph_type, winsize,