printing fix

Author: inikishev

visualbench/runs/benchpack.py

@@ -128,7 +128,7 @@ def logger_fn(value: float):
                     os.rename(f'{video_path} __TEMP__.mp4', f'{video_path}.mp4')
 
             if print_time:
-                if print_progress: print("\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t", end="\r")
+                if print_progress: print("                                                                                                  ", end="\r")
                 print(f"{task_name} took {(time.time() - start):.2f} s.")
 
         self.run_bench = run_bench

visualbench/runs/mlbench.py

@@ -82,11 +82,11 @@ def run_visual(self):
         # basic
         # ------------------------------ Rosenbrock-256 ------------------------------ #
         bench = tasks.projected.Rosenbrock(384).to(CUDA_IF_AVAILABLE)
-        self.run_bench(bench, 'S - Rosenbrock 384', passes=2000, sec=30, metrics='train loss', vid_scale=4, binary_mul=3)
+        self.run_bench(bench, 'S - Rosenbrock 384', passes=2_000, sec=30, metrics='train loss', vid_scale=4, binary_mul=3)
 
         # ------------------------------- neural drawer ------------------------------ #
         bench = tasks.NeuralDrawer(data.WEEVIL96, models.MLP([2,16,16,16,16,16,16,16,3], act_cls=nn.ReLU, bn=True), expand=48).to(CUDA_IF_AVAILABLE)
-        self.run_bench(bench, 'Visual - NeuralDrawer - ReLU+bn', passes=2000, sec=60, metrics='train loss', vid_scale=2, binary_mul=3)
+        self.run_bench(bench, 'Visual - NeuralDrawer - ReLU+bn', passes=2_000, sec=60, metrics='train loss', vid_scale=2, binary_mul=3)
 
         # ------------------------------- Colorization ------------------------------- #
         # ndim  = 1024