Need to save training checkpoints after several epoch so next training could start from this one
Need to save training checkpoints after several epoch so next training could start from this one