Use 'fork' as W&B start method

alex-snd · Sep 24, 2022 · 7f23cce · 7f23cce
1 parent d0ebee4
commit 7f23cce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/trecover/train/collab/monitor.py b/src/trecover/train/collab/monitor.py
@@ -58,7 +58,8 @@ def __init__(self,
                 id=wandb_id,
                 dir=wandb_registry,
                 resume='allow',
-                anonymous='never'
+                anonymous='never',
+                settings=wandb.Settings(start_method='fork')
             )
 
         self._peer_status()

diff --git a/src/trecover/train/collab/visualization.py b/src/trecover/train/collab/visualization.py
@@ -58,7 +58,8 @@ def __init__(self,
                 id=wandb_id,
                 dir=wandb_registry,
                 resume='allow',
-                anonymous='never'
+                anonymous='never',
+                settings=wandb.Settings(start_method='fork')
             )
 
     def stream(self) -> Generator[Tuple[int, List[Panel]], None, None]: