davjoython commited on
Commit
35335b2
·
verified ·
1 Parent(s): 3bff21b

Upload config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -4,7 +4,7 @@ research purposes. Don't try this code if you are a software engineer.'''
4
  # device settings
5
  device = 'cuda' # or 'cpu'
6
  import torch
7
- torch.cuda.set_device(0)
8
 
9
  # data settings
10
  dataset_path = "dummy_dataset"
 
4
  # device settings
5
  device = 'cuda' # or 'cpu'
6
  import torch
7
+ # torch.cuda.set_device(0)
8
 
9
  # data settings
10
  dataset_path = "dummy_dataset"