Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from logging import warn
import warnings
import torch
from torch.nn import Softmax
import numpy as np
Expand Down Expand Up @@ -33,7 +33,7 @@ def get_device(device):
return "mps"
if device == "cpu" or device == "best":
return "cpu"
warn(f"Requested device {device} not found, running on CPU")
warnings.warn(f"Requested device {device} not found, running on CPU")
return "cpu"


Expand Down