Spaces:
				
			
			
	
			
			
					
		Running
		
	
	
	
			
			
	
	
	
	
		
		
					
		Running
		
	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):
         |