mm/citybert
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

no mps

Browse Source
This commit is contained in:
Michael Pilosov 2026-05-25 14:26:37 -06:00
parent e8419354f4
commit bfad4547ce
3 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -3,3 +3,4 @@ plots*
*.csv *.csv
output/ output/
.requirements_installed .requirements_installed
__pycache__/

2
eval.py
View File

@ -38,8 +38,6 @@ def get_device(requested_device):
return torch.device(requested_device) return torch.device(requested_device)
if torch.cuda.is_available(): if torch.cuda.is_available():
return torch.device("cuda") return torch.device("cuda")
if torch.backends.mps.is_available():
return torch.device("mps")
return torch.device("cpu") return torch.device("cpu")

2
train.py
View File

@ -74,8 +74,6 @@ def get_device(requested_device):
return torch.device(requested_device) return torch.device(requested_device)
if torch.cuda.is_available(): if torch.cuda.is_available():
return torch.device("cuda") return torch.device("cuda")
if torch.backends.mps.is_available():
return torch.device("mps")
return torch.device("cpu") return torch.device("cpu")