updated args

This commit is contained in:
William Falcon 2019-06-25 18:18:40 -04:00
parent e58eee8d6a
commit e58cfafa74
1 changed files with 3 additions and 0 deletions

View File

@ -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')