Skip to content

Commit a3d1fd4

Browse files
committed
Fix pickling NeptuneLogger in offline mode
1 parent 8d0236f commit a3d1fd4

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pytorch_lightning/loggers/neptune.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -197,8 +197,12 @@ def __init__(self,
197197

198198
def __getstate__(self):
199199
state = self.__dict__.copy()
200-
# cannot be pickled
200+
201+
# Experiment cannot be pickled, and additionally its ID cannot be pickled in offline mode
201202
state['_experiment'] = None
203+
if self.offline_mode:
204+
state['_experiment_id'] = None
205+
202206
return state
203207

204208
@property
@@ -358,7 +362,8 @@ def _create_or_get_experiment(self):
358362
if self.offline_mode:
359363
project = neptune.Session(backend=neptune.OfflineBackend()).get_project('dry-run/project')
360364
else:
361-
project = neptune.Session.with_default_backend(api_token=self.api_key).get_project(self.project_name)
365+
session = neptune.Session.with_default_backend(api_token=self.api_key)
366+
project = session.get_project(self.project_name)
362367

363368
if self._experiment_id is None:
364369
exp = project.create_experiment(

0 commit comments

Comments
 (0)