Using the keras sequential utils as generator