train.py
file is a core building block in Slai. It’s used for:
- Writing training code for your model
- Loading pre-trained models
-
You need a
train()
function -
train()
must return a trained model, or an empty object
train()
function.
train()
, like so.
train.py
will be passed to the handler, which is a wrapper around the model object.