Spaces:
Runtime error
Runtime error
Upload config.py
Browse files
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"
|