Only save every 20 epochs