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:
run: |
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/PyTorchLightning/pytorch-lightning.git'
pip install git+https://github.com/facebookresearch/mobile-vision
pip install scikit-learn
pip install pytest
......@@ -31,4 +30,3 @@ jobs:
OSSRUN: 1
run: |
python -m unittest discover tests
......@@ -27,7 +27,7 @@ requirements = [
"Pillow",
"mock",
"torch",
"pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning.git@86b177ebe",
"pytorch-lightning @ git+https://github.com/PyTorchLightning/pytorch-lightning@15364c18c",
"opencv-python",
"parameterized",
"diskcache", # TODO: move to mobile_cv
......
......@@ -222,7 +222,7 @@ class TestLightningTask(unittest.TestCase):
assert batch
assert opt
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
# which is used to determine when to stop training
# 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