Training the CNTK model