import torch

cpu_device = torch.device("cpu")
# device = torch.device("cuda:3" if torch.cuda.is_available() else "cpu")
device = cpu_device