gpu usage

app.py

@@ -41,8 +41,18 @@ def cmd(command):
     print(command)
     subprocess.run(shlex.split(command))
 
-@spaces.GPU
-def cmd_gpu(command):
+@spaces.GPU(duration=70)
+def cmd_gpu_s1(command):
+    print('gpu:', command)
+    subprocess.run(shlex.split(command))
+
+@spaces.GPU(duration=40)
+def cmd_gpu_s2(command):
+    print('gpu:', command)
+    subprocess.run(shlex.split(command))
+
+@spaces.GPU(duration=20)
+def cmd_gpu_s3(command):
     print('gpu:', command)
     subprocess.run(shlex.split(command))
 
@@ -65,18 +75,18 @@ def process(inputfiles, input_path='demo'):
 
     imgs_path = temp_dir
     output_path = f'./results/{input_path}/output'
-    cmd_gpu(f"python dynamic_predictor/launch.py --mode=eval_pose_custom \
+    cmd_gpu_s1(f"python dynamic_predictor/launch.py --mode=eval_pose_custom \
         --pretrained=Kai422kx/das3r \
         --dir_path={imgs_path} \
         --output_dir={output_path} \
-        --use_pred_mask ")
+        --use_pred_mask --n_iter 150")
     
     cmd(f"python utils/rearrange.py --output_dir={output_path}")
     output_path = f'{output_path}_rearranged'
 
 
-    cmd_gpu(f"python train_gui.py -s {output_path} -m {output_path} --iter 2000")
-    cmd_gpu(f"python render.py -s {output_path} -m {output_path} --iter 2000 --get_video")
+    cmd_gpu_s2(f"python train_gui.py -s {output_path} -m {output_path} --iter 2000")
+    cmd_gpu_s3(f"python render.py -s {output_path} -m {output_path} --iter 2000 --get_video")
 
 
     output_video_path = f"{output_path}/rendered.mp4"