change kaggle pack

Author: inikishev

visualbench/runs/colab_utils.py

@@ -29,12 +29,6 @@ def colab_pack():
 
 def kaggle_pack():
     shutil.make_archive("MLBench", 'zip', "MLBench")
-    import os
-    os.chdir(r'/kaggle/working')
-    from IPython.display import FileLink
-    FileLink("MLBench.zip")
-
-
 
 def performance_tweaks(
     cudnn_bench: bool | None,

visualbench/runs/visualbench.py

@@ -85,7 +85,7 @@ def run_2d(self):
         bench = tasks.FunctionDescent('mycs1')
         self.run_bench(bench, '2D - mycs1', passes=500, sec=60, metrics='train loss', vid_scale=1)
 
-        bench = tasks.SimultaneousFunctionDescent('rosen').cuda()
+        bench = tasks.SimultaneousFunctionDescent('rosen').to(CUDA_IF_AVAILABLE)
         self.run_bench(bench, '2D simultaneous - rosenbrock', passes=1000, sec=60, metrics='train loss', vid_scale=3)
 
     def run_projected(self):
@@ -171,11 +171,13 @@ def run_visual(self):
         self.run_bench(bench, 'Visual - LinesDrawer SSIM', passes=2000, sec=60, metrics='train loss', vid_scale=4, fps=30)
 
         # -------------------------- deformable registration ------------------------- #
-        bench = tasks.DeformableRegistration(data.FROG96, grid_size=(5,5)).cuda()
+        bench = tasks.DeformableRegistration(data.FROG96, grid_size=(5,5)).to(CUDA_IF_AVAILABLE)
         self.run_bench(bench, 'Visual - DeformableRegistration', passes=2_000, sec=60, metrics='train loss', vid_scale=2)
 
     def run_linalg(self):
         # ---------------------------------- inverse --------------------------------- #
-        bench = tasks.Inverse(data.SANIC96).cuda()
+        bench = tasks.Inverse(data.SANIC96).to(CUDA_IF_AVAILABLE)
         self.run_bench(bench, 'Linalg - Inverse', passes=2_000, sec=60, metrics='train loss', vid_scale=2)
 
+        bench = tasks.StochasticInverse(data.SANIC96).to(CUDA_IF_AVAILABLE)
+        self.run_bench(bench, 'Linalg - StochasticInverse', passes=2_000, sec=60, metrics='train loss', vid_scale=2)