wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-17 11:32:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Tune, Ray SGD] Update PTL integrations (#11271)

Browse Source
This commit is contained in:
Amog Kamsetty
2020-10-08 13:43:07 -07:00
committed by GitHub
parent a6f91664c1
commit 1027bfd4b8
4 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_integration_pytorch_lightning.py
+2 -2
View File
@@ -46,8 +46,8 @@ class _MockModule(pl.LightningModule):
def validation_epoch_end(self, outputs):
avg_val_loss = torch.stack([x["val_loss"] for x in outputs]).mean()
avg_val_acc = torch.stack([x["val_acc"] for x in outputs]).mean()
return {"avg_val_loss": avg_val_loss, "avg_val_acc": avg_val_acc}
self.log("avg_val_loss", avg_val_loss)
self.log("avg_val_acc", avg_val_acc)
def configure_optimizers(self):
return None