mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
c0bedd2587
* Add allegro.ai TRAINS experiment manager support * improve docstring and type hinting, fix the bug in log_metrics, add support torch.Tensor to input into log_image * complete missing docstring of constructor's arguments * fix docs * pep8 * pep8 * remove redundant typing use logging fix typing and pep8 * remove deprecated interface * add TrainsLogger test * add TrainsLogger PR in CHANGELOG * add id/name property documentation * change logging as log Co-authored-by: bmartinn <> Co-authored-by: Sou Uchida <s.aiueo32@gmail.com>
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import pickle
|
|
|
|
import tests.models.utils as tutils
|
|
from pytorch_lightning import Trainer
|
|
from pytorch_lightning.loggers import TrainsLogger
|
|
from tests.models import LightningTestModel
|
|
|
|
|
|
def test_trains_logger(tmpdir):
|
|
"""Verify that basic functionality of TRAINS logger works."""
|
|
tutils.reset_seed()
|
|
|
|
hparams = tutils.get_hparams()
|
|
model = LightningTestModel(hparams)
|
|
logger = TrainsLogger(project_name="examples", task_name="pytorch lightning test")
|
|
|
|
trainer_options = dict(
|
|
default_save_path=tmpdir,
|
|
max_epochs=1,
|
|
train_percent_check=0.05,
|
|
logger=logger
|
|
)
|
|
trainer = Trainer(**trainer_options)
|
|
result = trainer.fit(model)
|
|
|
|
print('result finished')
|
|
assert result == 1, "Training failed"
|
|
|
|
|
|
def test_trains_pickle(tmpdir):
|
|
"""Verify that pickling trainer with TRAINS logger works."""
|
|
tutils.reset_seed()
|
|
|
|
# hparams = tutils.get_hparams()
|
|
# model = LightningTestModel(hparams)
|
|
|
|
logger = TrainsLogger(project_name="examples", task_name="pytorch lightning test")
|
|
|
|
trainer_options = dict(
|
|
default_save_path=tmpdir,
|
|
max_epochs=1,
|
|
logger=logger
|
|
)
|
|
|
|
trainer = Trainer(**trainer_options)
|
|
pkl_bytes = pickle.dumps(trainer)
|
|
trainer2 = pickle.loads(pkl_bytes)
|
|
trainer2.logger.log_metrics({"acc": 1.0})
|