wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

update info for GPU (#1021)

Browse Source
This commit is contained in:
Jirka Borovec
2020-03-12 12:36:46 -04:00
committed by GitHub
parent b4d4e489bf
commit ed0b890e75
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.run_local_tests.sh
+4
View File
@@ -1,5 +1,9 @@
#!/usr/bin/env bash
# install APEX, see https://github.com/NVIDIA/apex#linux
# to imitate SLURM set only single node
export SLURM_LOCALID=0
# use this to run tests
rm -rf _ckpt_*
rm -rf ./tests/save_dir*