<gh_stars>0
import torch

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

# input_nbr = 3
imsize = 224
# batch_size = 32
# lr = 0.0001
patience = 50
# start_epoch = 0
# epochs = 120
print_freq = 20
save_folder = 'models'
