--- title: PL Trainer keywords: fastai sidebar: home_sidebar summary: "Implementation of trainer for training PyTorch Lightning models." description: "Implementation of trainer for training PyTorch Lightning models." nb_path: "nbs/trainers/pl_trainer.ipynb" ---
Example
class Args:
def __init__(self):
self.data_dir = '/content/data'
self.min_rating = 4
self.num_negative_samples = 99
self.min_uc = 5
self.min_sc = 5
self.val_p = 0.2
self.test_p = 0.2
self.num_workers = 2
self.normalize = False
self.batch_size = 32
self.seed = 42
self.shuffle = True
self.pin_memory = True
self.drop_last = False
self.split_type = 'stratified'
args = Args()
from recohut.datasets.movielens import ML1mDataModule
ds = ML1mDataModule(**args.__dict__)
ds.prepare_data()
from recohut.models.nmf import NMF
model = NMF(n_items=ds.data.num_items, n_users=ds.data.num_users, embedding_dim=20)
pl_trainer(model, ds, max_epochs=5)