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

[tune] Make metrics parameter optional in pytorch lightning integration (#11402)

Browse Source
This commit is contained in:
Kai Fricke
2020-10-14 17:50:34 -07:00
committed by GitHub
parent 34191107a3
commit f7120d2a18
2 changed files with 34 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_integration_pytorch_lightning.py
+19 -5
View File
@@ -66,9 +66,22 @@ class PyTorchLightningIntegrationTest(unittest.TestCase):
def tearDown(self):
pass
def testReportCallback(self):
def testReportCallbackUnnamed(self):
def train(config):
module = _MockModule(10, 20)
module = _MockModule(10., 20.)
trainer = pl.Trainer(
max_epochs=1,
callbacks=[TuneReportCallback(on="validation_end")])
trainer.fit(module)
analysis = tune.run(train, stop={TRAINING_ITERATION: 1})
self.assertEqual(analysis.trials[0].last_result["avg_val_loss"],
10. * 1.1)
def testReportCallbackNamed(self):
def train(config):
module = _MockModule(10., 20.)
trainer = pl.Trainer(
max_epochs=1,
callbacks=[
@@ -81,14 +94,15 @@ class PyTorchLightningIntegrationTest(unittest.TestCase):
analysis = tune.run(train, stop={TRAINING_ITERATION: 1})
self.assertEqual(analysis.trials[0].last_result["tune_loss"], 10 * 1.1)
self.assertEqual(analysis.trials[0].last_result["tune_loss"],
10. * 1.1)
def testCheckpointCallback(self):
tmpdir = tempfile.mkdtemp()
self.addCleanup(lambda: shutil.rmtree(tmpdir))
def train(config):
module = _MockModule(10, 20)
module = _MockModule(10., 20.)
trainer = pl.Trainer(
max_epochs=1,
callbacks=[
@@ -115,7 +129,7 @@ class PyTorchLightningIntegrationTest(unittest.TestCase):
self.addCleanup(lambda: shutil.rmtree(tmpdir))
def train(config):
module = _MockModule(10, 20)
module = _MockModule(10., 20.)
trainer = pl.Trainer(
max_epochs=1,
callbacks=[