Converting Kaggle Training Notebooks to Shareable Code

blog_run-plant-from-package.py

from pytorch_lightning import Trainer
from kaggle_plantpatho.data import PlantPathologyDM
from kaggle_plantpatho.models import LitResnet, MultiPlantPathology

# create DataModule with training/validation split
dm = PlantPathologyDM(batch_size=98)
# initialize ResNet50 network
net = LitResnet(arch='resnet50', num_classes=dm.num_classes)
# initialize PL module with ResNet50
model = MultiPlantPathology(model=net, lr=6e-4)
# create Trainer with custome arguments
trainer = Trainer(...)
# train the model
trainer.fit(model=model, datamodule=dm)