Skip to content

Commit 6d1ee3b

Browse files
justusschocklexierule
authored andcommitted
Update csv_logs.py
1 parent 51cb077 commit 6d1ee3b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_lightning/loggers/csv_logs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ def save(self) -> None:
9595
metrics_keys = list(last_m.keys())
9696

9797
with open(self.metrics_file_path, "w", newline="") as f:
98+
# Don't assign the writer to self.
99+
# Keeps an open reference and prevents pickling otherwise
98100
writer = csv.DictWriter(f, fieldnames=metrics_keys)
99101
writer.writeheader()
100102
writer.writerows(self.metrics)

0 commit comments

Comments
 (0)