from trainer import fit
from model import model
import torch
from data_loader import train_loader, val_loader

lr = 0.001
num_epochs = 20
opt_func = torch.optim.Adam

# Commented out IPython magic to ensure Python compatibility.
# history = fit(num_epochs, lr, model, train_loader, val_loader, opt_func)

# torch.save(model.state_dict(), 'potato_model_statedict__f.pth')