From a3b4d5e76e10f0817588d7301f2e52be32b84fd0 Mon Sep 17 00:00:00 2001 From: Yige Xu Date: Wed, 11 Sep 2019 14:10:01 +0800 Subject: [PATCH] remove tensorboard logs after testing TensorboardCallback --- test/core/test_callbacks.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/test/core/test_callbacks.py b/test/core/test_callbacks.py index 909295c0..5fc8cbf8 100644 --- a/test/core/test_callbacks.py +++ b/test/core/test_callbacks.py @@ -85,6 +85,9 @@ class TestCallback(unittest.TestCase): metrics=AccuracyMetric(pred="predict", target="y"), use_tqdm=False, callbacks=[TensorboardCallback("loss", "metric")], check_code_level=2) trainer.train() + import os + import shutil + shutil.rmtree(os.path.join("./", 'tensorboard_logs_{}'.format(trainer.start_time))) def test_readonly_property(self): from fastNLP.core.callback import Callback