import torch

DATA_DIR = './resources/malimg_paper_dataset_imgs/'

MODEL_DIR = './model_dict'

NUM_CLASS = 20

USE_GPU = torch.cuda.is_available()

DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
