From d6b5f37a7b553ab4b1254dad7ecae1eefb4a6cd6 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 27 Jul 2019 14:28:44 -0400 Subject: [PATCH] Update README.md --- README.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 315de45f..d2488324 100644 --- a/README.md +++ b/README.md @@ -99,13 +99,18 @@ class CoolModel(ptl.LightningModule): from pytorch_lightning import Trainer from test_tube import Experiment -model = CoolModel() - -# fit on 32 gpus across 4 nodes model = CoolModel() exp = Experiment(save_dir=os.getcwd()) + +# train on cpu trainer = Trainer(experiment=exp, max_nb_epochs=1) +# train on 4 gpus +# trainer = Trainer(experiment=exp, max_nb_epochs=1, gpus=[0, 1, 2, 3]) + +# train on 32 gpus across 4 nodes (make sure to submit appropriate SLURM job) +# trainer = Trainer(experiment=exp, max_nb_epochs=1, gpus=[0, 1, 2, 3, 4, 5, 6, 7], nb_gpu_nodes=4) + # train (1 epoch only here for demo) trainer.fit(model)