From 20be3877b292d8d5b9228c389b667f84300cbbb0 Mon Sep 17 00:00:00 2001 From: golmschenk Date: Thu, 7 Mar 2024 13:54:19 -0500 Subject: [PATCH] Switch from wandb start mode from fork to thread --- src/qusi/single_dense_layer_model.py | 3 +-- src/qusi/train_session.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/qusi/single_dense_layer_model.py b/src/qusi/single_dense_layer_model.py index fed8ffe9..f5a070e2 100644 --- 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 index 405f3bb2..28f41ed3 100644 --- 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)