mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-23 13:40:39 +08:00
Jirka Borovec
45d671a4a8
* CI: split tests-examples * tests without template * comment depends * CircleCI typo * add doctest * update test req. * CI tests * setup macOS * longer train * lover pred acc * fix model * rename default model * lower tests acc * typo * imports * fix test optimizer * update calls * fix Win * lower Drone image * fix call * pytorch image * fix test * add dev image * add dev image * update image * drone volume * lint * update test notes * rename tests/models >> tests/base * group models * conftest * optim imports * typos * fix import * fix tests * install AMP * tests * fix import
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
import pickle
|
|
|
|
import tests.base.utils as tutils
|
|
from pytorch_lightning import Trainer
|
|
from pytorch_lightning.loggers import TrainsLogger
|
|
from tests.base import LightningTestModel
|
|
|
|
|
|
def test_trains_logger(tmpdir):
|
|
"""Verify that basic functionality of TRAINS logger works."""
|
|
tutils.reset_seed()
|
|
|
|
hparams = tutils.get_default_hparams()
|
|
model = LightningTestModel(hparams)
|
|
TrainsLogger.set_bypass_mode(True)
|
|
TrainsLogger.set_credentials(api_host='http://integration.trains.allegro.ai:8008',
|
|
files_host='http://integration.trains.allegro.ai:8081',
|
|
web_host='http://integration.trains.allegro.ai:8080', )
|
|
logger = TrainsLogger(project_name="lightning_log", 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')
|
|
logger.finalize()
|
|
assert result == 1, "Training failed"
|
|
|
|
|
|
def test_trains_pickle(tmpdir):
|
|
"""Verify that pickling trainer with TRAINS logger works."""
|
|
tutils.reset_seed()
|
|
|
|
# hparams = tutils.get_default_hparams()
|
|
# model = LightningTestModel(hparams)
|
|
TrainsLogger.set_bypass_mode(True)
|
|
TrainsLogger.set_credentials(api_host='http://integration.trains.allegro.ai:8008',
|
|
files_host='http://integration.trains.allegro.ai:8081',
|
|
web_host='http://integration.trains.allegro.ai:8080', )
|
|
logger = TrainsLogger(project_name="lightning_log", 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})
|
|
trainer2.logger.finalize()
|
|
logger.finalize()
|