Files
pytorch-lightning/environment.yml
T
c0bedd2587 Add TRAINS experiment manager support (#1122)
* 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>
2020-03-14 13:02:14 -04:00

36 lines
652 B
YAML

# Usage: `conda env update -f environment.yml`
channels:
- conda-forge
- pytorch
dependencies:
- python==3.7.6
- pip==20.0.2
- tqdm>=4.35.0
- numpy>=1.16.4
- pytorch>=1.1
- tensorboard>=1.14
- future>=0.17.1
# For dev and testing
- torchvision>=0.4.0
- tox
- coverage
- codecov
- pytest>=3.0.5
- pytest-cov
- pytest-flake8
- flake8
- autopep8
- check-manifest
- twine==1.13.0
- pillow<7.0.0
- pip:
- test-tube>=0.7.5
- mlflow>=1.0.0
- comet_ml>=1.0.56
- wandb>=0.8.21
- neptune-client>=0.4.4
- trains>=0.13.3