Update config.py
Browse files
config.py
CHANGED
|
@@ -6,7 +6,7 @@ import os
|
|
| 6 |
from pathlib import Path
|
| 7 |
from types import SimpleNamespace
|
| 8 |
|
| 9 |
-
project_root = Path(__file__).resolve().parent
|
| 10 |
|
| 11 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 12 |
|
|
|
|
| 6 |
from pathlib import Path
|
| 7 |
from types import SimpleNamespace
|
| 8 |
|
| 9 |
+
project_root = Path(__file__).resolve().parent
|
| 10 |
|
| 11 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 12 |
|