Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add exit when GPU not found

Browse Source
This commit is contained in:
Yin Li 2020-04-20 19:27:57 -04:00
parent facfc13132
commit 5adf6cbac6
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
map2map/train.py
View File

@ -35,6 +35,9 @@ def node_worker(args):
node = int(os.environ['SLURM_NODEID']) node = int(os.environ['SLURM_NODEID'])
if args.gpus_per_node < 1:
raise RuntimeError('GPU not found on node {}'.format(node))
spawn(gpu_worker, args=(node, args), nprocs=args.gpus_per_node) spawn(gpu_worker, args=(node, args), nprocs=args.gpus_per_node)