Should "if ( _ / update_epochs == 0):" be "if ( _ % update_epochs == 0):"?
Should "if ( _ / update_epochs == 0):" be "if ( _ % update_epochs == 0):"?