Commit 096ab230 authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

bump up lightning version

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/187

Reviewed By: ananthsub, zhanghang1989

Differential Revision: D34650467

fbshipit-source-id: b9518e5dd673b709320b87e57a43d053eca3aabe
parent 86e2dc7f
...@@ -20,7 +20,6 @@ jobs: ...@@ -20,7 +20,6 @@ jobs:
run: | run: |
conda install pytorch torchvision -c pytorch-nightly conda install pytorch torchvision -c pytorch-nightly
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
python -m pip install 'git+https://github.com/PyTorchLightning/pytorch-lightning.git'
pip install git+https://github.com/facebookresearch/mobile-vision pip install git+https://github.com/facebookresearch/mobile-vision
pip install scikit-learn pip install scikit-learn
pip install pytest pip install pytest
...@@ -31,4 +30,3 @@ jobs: ...@@ -31,4 +30,3 @@ jobs:
OSSRUN: 1 OSSRUN: 1
run: | run: |
python -m unittest discover tests python -m unittest discover tests
...@@ -27,7 +27,7 @@ requirements = [ ...@@ -27,7 +27,7 @@ requirements = [
"Pillow", "Pillow",
"mock", "mock",
"torch", "torch",
"pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning.git@86b177ebe", "pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@15364c18c",
"opencv-python", "opencv-python",
"parameterized", "parameterized",
"diskcache", # TODO: move to mobile_cv "diskcache", # TODO: move to mobile_cv
......
...@@ -222,7 +222,7 @@ class TestLightningTask(unittest.TestCase): ...@@ -222,7 +222,7 @@ class TestLightningTask(unittest.TestCase):
assert batch assert batch
assert opt assert opt
assert manual_backward assert manual_backward
# We take step the optimizer for progress tracking to occur # We step the optimizer for progress tracking to occur
# This is reflected in the Trainer's global_step property # This is reflected in the Trainer's global_step property
# which is used to determine when to stop training # which is used to determine when to stop training
# when specifying the loop bounds with Trainer(max_steps=N) # when specifying the loop bounds with Trainer(max_steps=N)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment