FastAI’s callbacks for better CNN training — meet SaveModelCallback.

Less Wright
3 min readJun 25, 2019

FastAI has a very flexible callback system that let’s you greatly customize your training process. However, some of the pre-built and useful callbacks are not as easy to find without a deep dive into the documentation and to my knowledge, aren’t covered in the regular courses.

A common question is thus, “how do I automatically save my best model if it happens in the middle of a training run?” and the answer is to use the SaveModelCallback.

--

--

Less Wright

PyTorch, Deep Learning, Object detection, Stock Index investing and long term compounding.