| @@ -134,7 +134,7 @@ class TestCallback(unittest.TestCase): | |||||
| def test_fitlog_callback(self): | def test_fitlog_callback(self): | ||||
| import fitlog | import fitlog | ||||
| fitlog.set_log_dir(self.tempdir) | |||||
| fitlog.set_log_dir(self.tempdir, new_log=True) | |||||
| data_set, model = prepare_env() | data_set, model = prepare_env() | ||||
| from fastNLP import Tester | from fastNLP import Tester | ||||
| tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) | tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) | ||||
| @@ -164,7 +164,7 @@ class TestCallback(unittest.TestCase): | |||||
| tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) | tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) | ||||
| import fitlog | import fitlog | ||||
| fitlog.set_log_dir(self.tempdir) | |||||
| fitlog.set_log_dir(self.tempdir, new_log=True) | |||||
| tempfile_path = os.path.join(self.tempdir, 'chkt.pt') | tempfile_path = os.path.join(self.tempdir, 'chkt.pt') | ||||
| callbacks = [CheckPointCallback(tempfile_path)] | callbacks = [CheckPointCallback(tempfile_path)] | ||||