Browse Source

remove tensorboard logs after testing TensorboardCallback

tags/v0.4.10
Yige Xu 5 years ago
parent
commit
a3b4d5e76e
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      test/core/test_callbacks.py

+ 3
- 0
test/core/test_callbacks.py View File

@@ -85,6 +85,9 @@ class TestCallback(unittest.TestCase):
metrics=AccuracyMetric(pred="predict", target="y"), use_tqdm=False, metrics=AccuracyMetric(pred="predict", target="y"), use_tqdm=False,
callbacks=[TensorboardCallback("loss", "metric")], check_code_level=2) callbacks=[TensorboardCallback("loss", "metric")], check_code_level=2)
trainer.train() trainer.train()
import os
import shutil
shutil.rmtree(os.path.join("./", 'tensorboard_logs_{}'.format(trainer.start_time)))
def test_readonly_property(self): def test_readonly_property(self):
from fastNLP.core.callback import Callback from fastNLP.core.callback import Callback


Loading…
Cancel
Save