File size: 370 Bytes
32c431e |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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') |