File tree 1 file changed +7
-2
lines changed
pytorch_lightning/loggers
1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -197,8 +197,12 @@ def __init__(self,
197
197
198
198
def __getstate__ (self ):
199
199
state = self .__dict__ .copy ()
200
- # cannot be pickled
200
+
201
+ # Experiment cannot be pickled, and additionally its ID cannot be pickled in offline mode
201
202
state ['_experiment' ] = None
203
+ if self .offline_mode :
204
+ state ['_experiment_id' ] = None
205
+
202
206
return state
203
207
204
208
@property
@@ -358,7 +362,8 @@ def _create_or_get_experiment(self):
358
362
if self .offline_mode :
359
363
project = neptune .Session (backend = neptune .OfflineBackend ()).get_project ('dry-run/project' )
360
364
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 )
362
367
363
368
if self ._experiment_id is None :
364
369
exp = project .create_experiment (
You can’t perform that action at this time.
0 commit comments