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

fix deprecated tng and abstract ligntning (#644)

Browse Source
This commit is contained in:
Jirka Borovec
2020-01-13 22:20:38 -05:00
committed by William Falcon
parent 8dc8a8bfd3
commit f7db44e750
4 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_trainer.py
+2 -2
View File
@@ -15,7 +15,7 @@ from pytorch_lightning.testing import (
LightningValidationMultipleDataloadersMixin,
LightningTestMultipleDataloadersMixin,
)
from pytorch_lightning.trainer import trainer_io
from pytorch_lightning.trainer import training_io
from pytorch_lightning.trainer.logging import TrainerLoggingMixin
@@ -187,7 +187,7 @@ def test_loading_meta_tags(tmpdir):
tags_path = logger.experiment.get_data_path(
logger.experiment.name, logger.experiment.version
) + '/meta_tags.csv'
tags = trainer_io.load_hparams_from_tags_csv(tags_path)
tags = training_io.load_hparams_from_tags_csv(tags_path)
assert tags.batch_size == 32 and tags.hidden_dim == 1000