Auto-format by https://ultralytics.com/actions

ultralytics · Oct 15, 2024 · 6fb9b9d · 6fb9b9d
1 parent 8519789
commit 6fb9b9d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 18 deletions.
diff --git a/gcp/wave_pytorch_gcp.py b/gcp/wave_pytorch_gcp.py
@@ -28,17 +28,11 @@ def runexample(H, model, str, lr=0.001, amsgrad=False):
 
     cuda = torch.cuda.is_available()
     os.makedirs(f"{pathr}models", exist_ok=True)
-    name = (
-        f"{data[:-4]}{H[:]}{lr:g}lr{str}".replace(", ", ".")
-        .replace("[", "_")
-        .replace("]", "_")
-    )
+    name = f"{data[:-4]}{H[:]}{lr:g}lr{str}".replace(", ", ".").replace("[", "_").replace("]", "_")
 
     tica = time.time()
     device = torch.device("cuda:0" if cuda else "cpu")
-    print(
-        f'Running {name} on {device.type}\n{torch.cuda.get_device_properties(0) if cuda else ""}'
-    )
+    print(f'Running {name} on {device.type}\n{torch.cuda.get_device_properties(0) if cuda else ""}')
 
     if not os.path.isfile(pathd + data):
         os.system(f"wget -P data/ https://storage.googleapis.com/ultralytics/{data}")

diff --git a/train.py b/train.py
@@ -23,11 +23,7 @@ def train(H, model, str, lr=0.001):
 
     cuda = torch.cuda.is_available()
     os.makedirs(f"{pathr}models", exist_ok=True)
-    name = (
-        f"{data[:-4]}{H[:]}{lr:g}lr{str}".replace(", ", ".")
-        .replace("[", "_")
-        .replace("]", "_")
-    )
+    name = f"{data[:-4]}{H[:]}{lr:g}lr{str}".replace(", ", ".").replace("[", "_").replace("]", "_")
     print(f"Running {name}")
 
     device = select_device()

diff --git a/train_tf.py b/train_tf.py
@@ -25,11 +25,7 @@ def runexample(H, model, str):
     tf.set_random_seed(1)
     path = "data/"
     os.makedirs(f"{path}models", exist_ok=True)
-    name = (
-        f"{data[:-4]}{H[:]}{lr:g}lr{eps:g}eps{str}".replace(", ", "_")
-        .replace("[", "_")
-        .replace("]", "_")
-    )
+    name = f"{data[:-4]}{H[:]}{lr:g}lr{eps:g}eps{str}".replace(", ", "_").replace("[", "_").replace("]", "_")
 
     tica = time.time()
     device = "/gpu:0" if cuda else "/cpu:0"