mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-01 12:32:04 +08:00
Adrian WälchliandJirka Borovec
b7de42f70d
* added custom mnist without torchvision dep * move files so it does not conflict with mnist gitignore * mock torchvision for tests * fix line too long * fix line too long * fix "module level import not at top of file" warning * move mock imports to __init__.py * simplify MNIST a lot and download directly the .pt files * further simplify and clean up mnist * revert import overrides * make as before * drop PIL requirement * move mnist.py to datasets subfolder * use logging instead of print * choose same name as in torchvision * remove torchvision and pillow also from yml file * refactor if train Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * capitalized class attr * moved mnist to models * re-added datsets ignore * better name for file variable * Update mnist.py * move dataset classes to datasets.py * new line * update * update * fix automerge * move to base folder * adapt testingmnist to new mnist base class * remove temporal fix * fix datatype * remove old testingmnist * readable * fix import * fix whitespace * docstring Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * Update tests/base/datasets.py Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * changelog * added types * Update CHANGELOG.md Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * exist->isfile Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com> * index -> idx * temporary fix for trains error * better changelog message Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
36 lines
642 B
YAML
36 lines
642 B
YAML
# This is Conda environment file
|
|
# 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
|
|
- tox
|
|
- coverage
|
|
- codecov
|
|
- pytest>=3.0.5
|
|
- pytest-cov
|
|
- pytest-flake8
|
|
- flake8
|
|
- autopep8
|
|
- check-manifest
|
|
- twine==1.13.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
|