wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 20:53:14 +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/integration/pytorch_lightning.py
+3 -3
View File
@@ -78,8 +78,8 @@ class TuneCallback(Callback):
self._handle(trainer, pl_module)
def on_validation_batch_end(self, trainer: Trainer,
pl_module: LightningModule, batch, batch_idx,
dataloader_idx):
pl_module: LightningModule, outputs, batch,
batch_idx, dataloader_idx):
if "validation_batch_end" in self._on:
self._handle(trainer, pl_module)
@@ -89,7 +89,7 @@ class TuneCallback(Callback):
self._handle(trainer, pl_module)
def on_test_batch_end(self, trainer: Trainer, pl_module: LightningModule,
batch, batch_idx, dataloader_idx):
outputs, batch, batch_idx, dataloader_idx):
if "test_batch_end" in self._on:
self._handle(trainer, pl_module)
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):