Increase time, add test line to check if cuda() calls are legal

demo/gs_train.py

@@ -60,7 +60,7 @@ class TrainingArgs:
     checkpoint_iterations: list[int] = field(default_factory=lambda: [7_000, 15_000, 30_000])
     start_checkpoint: str = None
 
-@spaces.GPU(duration=10)
+@spaces.GPU(duration=90)
 def train(
     data_source_path, sh_degree, model_path, images, resolution, white_background, data_device, eval,
     convert_SHs_python, compute_cov3D_python, debug,
@@ -241,7 +241,7 @@ def train(
         iteration = scene.loaded_iter
 
         bg_color = [1,1,1] if dataset.white_background else [0, 0, 0]
-        background = torch.tensor(bg_color, dtype=torch.float32, device="cuda")
+        background = torch.tensor(bg_color, dtype=torch.float32, device=DEVICE)
 
         model_path = dataset.model_path
         name = "render"

demo/mast3r_demo.py

@@ -213,6 +213,8 @@ def get_reconstructed_scene(image_size, current_scene_state,
             run_counter += 1
         return run_cache_dir
 
+
+    ten = torch.zeros((1024)).cuda()
     cache_dir = get_next_dir(base_cache_dir)
     scene = sparse_global_alignment(filelist, pairs, cache_dir,
                                     MODEL, lr1=lr1, niter1=niter1, lr2=lr2, niter2=niter2, device=DEVICE,