We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cb9dac9 commit b29d898Copy full SHA for b29d898
code/DeepDG/train.py
@@ -90,7 +90,7 @@ def get_args():
90
args = parser.parse_args()
91
args.steps_per_epoch = 100
92
args.data_dir = args.data_file+args.data_dir
93
- os.environ['CUDA_VISIBLE_DEVICS'] = args.gpu_id
+ os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_id
94
os.makedirs(args.output, exist_ok=True)
95
sys.stdout = Tee(os.path.join(args.output, 'out.txt'))
96
sys.stderr = Tee(os.path.join(args.output, 'err.txt'))
0 commit comments