Switch from wandb start mode from fork to thread

golmschenk · Mar 7, 2024 · 20be387 · 20be387
1 parent e363126
commit 20be387
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/qusi/single_dense_layer_model.py b/src/qusi/single_dense_layer_model.py
@@ -1,10 +1,9 @@
 from __future__ import annotations
 
-from typing_extensions import Self
-
 import torch
 from torch import Tensor
 from torch.nn import Linear, Module, Sigmoid
+from typing_extensions import Self
 
 
 class SingleDenseLayerBinaryClassificationModel(Module):

diff --git a/src/qusi/train_session.py b/src/qusi/train_session.py
@@ -37,7 +37,7 @@ def train_session(
         process_rank=0,
         project=logging_configuration.wandb_project,
         entity=logging_configuration.wandb_entity,
-        settings=wandb.Settings(start_method="fork"),
+        settings=wandb.Settings(start_method="thread"),
     )
     sessions_directory = Path("sessions")
     sessions_directory.mkdir(exist_ok=True)