import torch
import torchvision
import torchvision.transforms as transforms

num_epochs = 10
batch_size = 32
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')