fix num iter

app.py

@@ -59,7 +59,7 @@ def get_dust3r_args_parser():
     return parser
 
 
-@spaces.GPU(duration=180)
+@spaces.GPU(duration=250)
 def run_dust3r(inputfiles, input_path=None):
 
     if input_path is not None:
@@ -154,8 +154,8 @@ def run_dust3r(inputfiles, input_path=None):
 
     return outfile, feat_image_path, opt, None, None
 
-@spaces.GPU(duration=180)
-def run_feat2gs(opt, niter=200):
+@spaces.GPU(duration=250)
+def run_feat2gs(opt, niter=2000):
 
     if opt is None:
         raise gr.Error("Please run Step 1 first!")
@@ -214,7 +214,7 @@ def run_feat2gs(opt, niter=200):
     except Exception as e:
         raise gr.Error(f"Step 2 failed: {str(e)}")
 
-@spaces.GPU(duration=180)
+@spaces.GPU(duration=250)
 def run_render(opt, args, cam_traj='ellipse'):
     if opt is None or args is None:
         raise gr.Error("Please run Steps 1 and 2 first!")
@@ -249,11 +249,11 @@ def run_render(opt, args, cam_traj='ellipse'):
     except Exception as e:
         raise gr.Error(f"Step 3 failed: {str(e)}")
 
-
+@spaces.GPU(duration=250)
 def process_example(inputfiles, input_path):
     dust3r_model, feat_image, dust3r_state, _, _ = run_dust3r(inputfiles, input_path=input_path)
     
-    output_model, feat2gs_state, _ = run_feat2gs(dust3r_state, niter=200)
+    output_model, feat2gs_state, _ = run_feat2gs(dust3r_state, niter=2000)
     
     output_video = run_render(dust3r_state, feat2gs_state, cam_traj='interpolated')
     
@@ -355,7 +355,7 @@ with demo:
 
     with gr.Row(variant='panel'):
         with gr.Tab("Step 2: Readout 3DGS from features & Jointly optimize pose"):
-            niter = gr.Number(value=200, precision=0, minimum=0, maximum=8000, label="Training iterations")
+            niter = gr.Number(value=2000, precision=0, minimum=2000, maximum=8000, label="Training iterations")
             feat2gs_run = gr.Button("RUN Step 2")
             with gr.Column(scale=1):
                 with gr.Group():

train_feat2gs.py

@@ -66,7 +66,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi
     progress_bar = tqdm(range(first_iter, opt.iterations), desc="Training progress")
     first_iter += 1
 
-    warm_iter = 100
+    warm_iter = 1000
 
     start = perf_counter()
     for iteration in range(first_iter, opt.iterations + 1):