config.py

import torch

import util


device = (
    'cuda:0' if torch.cuda.is_available()
    else (
        'mps' if util.has_mps()
        else 'cpu'
    )
)

x_pad = 3 if util.is_half(device) else 1
x_query = 10 if util.is_half(device) else 6
x_center = 60 if util.is_half(device) else 38
x_max = 65 if util.is_half(device) else 41