Spaces:
Paused
Paused
Commit
·
9110acb
1
Parent(s):
d235373
Update main.py
Browse files
main.py
CHANGED
|
@@ -4,7 +4,7 @@ import logging
|
|
| 4 |
|
| 5 |
logger = logging.getLogger()
|
| 6 |
|
| 7 |
-
device = torch.device("cuda" if torch.cuda.is_available() else "
|
| 8 |
|
| 9 |
logger.info("Getting Device")
|
| 10 |
logger.info(device)
|
|
|
|
| 4 |
|
| 5 |
logger = logging.getLogger()
|
| 6 |
|
| 7 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 8 |
|
| 9 |
logger.info("Getting Device")
|
| 10 |
logger.info(device)
|