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

Add TRAINS experiment manager support (#1122)

Browse Source 
* 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>
This commit is contained in:
Martin.B
2020-03-14 13:02:14 -04:00
committed by GitHub
co-authored by bmartinn Sou Uchida
parent da61398835
commit c0bedd2587
9 changed files with 372 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/experiment_reporting.rst
+1 -1
View File
@@ -32,7 +32,7 @@ want to log using this trainer flag.
Log metrics
^^^^^^^^^^^
To plot metrics into whatever logger you passed in (tensorboard, comet, neptune, etc...)
To plot metrics into whatever logger you passed in (tensorboard, comet, neptune, TRAINS, etc...)
1. training_epoch_end, validation_epoch_end, test_epoch_end will all log anything in the "log" key of the return dict.