AutoTrain - Part 2: Recipes
[We have seen how to setup the ingredients for an experiment]({% post_url 2021-10-07-auto-train-config %}). The next step is to create the necessary files to use these ingredients in a way that training and validation can be done in one line.
All the recipes can be summarized into the following files
config.ini --> Ingredients
custom_functions.py --> User inputs that suppliment config.ini
model.py --> Base model (and optional dataloaders + learner during training)
train.py --> Training logic
validate.py --> Validation logic
main.py --> Command line and API logic
Let's create the recepie for MNIST ingredients that we have created in the previous article.
architecture
The first two files were already discussed in previous post. The common base for training and validation is the model itself. It can be then loaded into train / validate files and do the needed tasks. Let's start there.