Update test_tensorboard.py

2020-04-27 09:50:01 +02:00 · 2020-04-27 09:50:01 +02:00 · 312e394654
parent cdbf2f4a37
commit 312e394654
1 changed files with 16 additions and 0 deletions
--- a/tests/loggers/test_tensorboard.py
+++ b/tests/loggers/test_tensorboard.py
@ -77,3 +77,19 @@ def test_tensorboard_log_hyperparams(tmpdir):
        "layer": torch.nn.BatchNorm1d
    }
    logger.log_hyperparams(hparams)
+    
+def test_tensorboard_log_hparams_and_metrics
+    logger = TensorBoardLogger(tmpdir)
+    hparams = {
+        "float": 0.3,
+        "int": 1,
+        "string": "abc",
+        "bool": True,
+        "dict": {'a': {'b': 'c'}},
+        "list": [1, 2, 3],
+        "namespace": Namespace(foo=Namespace(bar='buzz')),
+        "layer": torch.nn.BatchNorm1d
+    }
+    metrics = {'abc': torch.tensor([0.54])}
+    logger.log_hyperparams(hparams, metrics)
+