lr = 2e-4, batch_size = 4, num_epochs = 10, weight_decay = 0.05

Upload model

initial commit