num_epochs = 8 opt_func = torch.optim.Adam lr = 1e-2 history = fit(num_epochs,lr,model, train_dl, val_dl,opt_func)