From e58cfafa74ac19d9e6b04790556956b311879407 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 18:18:40 -0400 Subject: [PATCH] updated args --- docs/source/examples/fully_featured_trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docs/source/examples/fully_featured_trainer.py b/docs/source/examples/fully_featured_trainer.py index c6cb3833fb..ab346b47fb 100644 --- a/docs/source/examples/fully_featured_trainer.py +++ b/docs/source/examples/fully_featured_trainer.py @@ -3,6 +3,7 @@ import sys import numpy as np from time import sleep import torch +import pdb from test_tube import HyperOptArgumentParser, Experiment, SlurmCluster from pytorch_lightning.models.trainer import Trainer @@ -171,6 +172,8 @@ if __name__ == '__main__': os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # RUN TRAINING + pdb.set_trace() + if hyperparams.on_cluster: # run on HPC cluster print('RUNNING ON SLURM CLUSTER')