wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix val logging (#362)

Browse Source 
* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* fix test

* no warnings always

* no warnings always

* no warnings always

* no warnings always
This commit is contained in:
William Falcon
2019-10-15 12:44:20 -04:00
committed by GitHub
parent b8666bf354
commit e2cabb03ba
4 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_logging.py
+5
View File
@@ -66,6 +66,8 @@ def test_testtube_pickle():
trainer2 = pickle.loads(pkl_bytes)
trainer2.logger.log_metrics({"acc": 1.0})
clear_save_dir()
def test_mlflow_logger():
"""
@@ -134,6 +136,9 @@ def test_mlflow_pickle():
trainer2 = pickle.loads(pkl_bytes)
trainer2.logger.log_metrics({"acc": 1.0})
n = np.random.randint(0, 10000000, 1)[0]
shutil.move(mlflow_dir, mlflow_dir + f'_{n}')
def test_custom_logger():
tests/test_models.py
+1
View File
@@ -1519,6 +1519,7 @@ def get_model(use_test_model=False, lbfgs=False):
hparams = get_hparams()
if lbfgs:
setattr(hparams, 'optimizer_name', 'lbfgs')
setattr(hparams, 'learning_rate', 0.002)
if use_test_model:
model = LightningTestModel(hparams)