wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-24 13:20:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Tune] Changes for Pytorch Lightning 1.0 (#11375)

Browse Source
This commit is contained in:
Amog Kamsetty
2020-10-13 15:50:11 -07:00
committed by GitHub
parent a6a94d3206
commit 933cf6675c
3 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_integration_pytorch_lightning.py
+1 -1
View File
@@ -34,7 +34,7 @@ class _MockModule(pl.LightningModule):
def forward(self, *args, **kwargs):
return self.loss
def backward(self, trainer, loss, optimizer, optimizer_idx):
def backward(self, loss, optimizer, optimizer_idx):
return None
def training_step(self, train_batch, batch_idx):