From a5168973c1a1154b3b4142fd056877fed09fe0d8 Mon Sep 17 00:00:00 2001 From: AleksanderWWW Date: Wed, 17 Jan 2024 22:15:57 +0100 Subject: [PATCH 1/2] tensor logging --- src/neptune_tensorboard/sync/sync_impl.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/neptune_tensorboard/sync/sync_impl.py b/src/neptune_tensorboard/sync/sync_impl.py index 7ccb479..a4ad093 100644 --- a/src/neptune_tensorboard/sync/sync_impl.py +++ b/src/neptune_tensorboard/sync/sync_impl.py @@ -4,6 +4,7 @@ import click import neptune +from neptune.utils import stringify_unsupported from tensorboard.backend.event_processing.event_accumulator import EventAccumulator try: @@ -102,5 +103,13 @@ def _export_to_neptune_run(self, path): for hparam in reader.hparams.itertuples(): namespace_handler["hparams"][hparam.tag].append(hparam.value) + # Read tensors + if hasattr(reader, "tensors"): + for tensor in reader.tensors.itertuples(): + namespace_handler["tensor"][tensor.tag].append( + stringify_unsupported(tensor.value), + step=tensor.step, + ) + # user facing click.echo(f"{path} was exported with run_id: {hash_run_id}") From e8d03b9f78f1894c5649f749640b68f73e04394a Mon Sep 17 00:00:00 2001 From: AleksanderWWW Date: Wed, 17 Jan 2024 22:28:56 +0100 Subject: [PATCH 2/2] changelog updated --- CHANGELOG.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index ea8d268..c73cd15 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,9 @@ +## [UNRELEASED] neptune-tensorboard 1.0.3 + +### Fixes +- Fix not logging `tensors` events ([#71](https://github.com/neptune-ai/neptune-tensorboard/pull/71)) + + ## neptune-tensorboard 1.0.2 ### Changes