Deep Learning, Pytorch, Fine Tuning, Computer Vision, Transfer Learning.
optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.9, weight_decay=0.001) # Define your optimizer with weight decay optimizer = optim.Adam(model.parameters(), lr=0.001, weight_decay=0.0001) Choosing Loss functions Image source : naptune.ai To explore and select a suitable loss function