import os
from pathlib import Path
from torchvision import transforms as trans

batch_size = 64
epochs = 10
lr = 0.1
momentum = 0.9
save_model = True
num_workers = 2
input_size = 112
embedding_size = 512

