Source code for wimarka.utils.torch

import torch



def get_device():
    return torch.device("cuda" if torch.cuda.is_available() else "cpu")




def move_model_to_device(model, device=None):
    if device is None:
        device = get_device()
    return model.to(device)