import torch device = 0 if torch.cuda.is_available() else -1 cache_dir = "./models_cache"