How about the fit_generator of Keras

[lancexiang]

Hi, have to say, it's a awesome tools for pytorch!

Will you continue to fix this for more convincent way to fit model? Just like the Keras: model.fit_generator, which is faster.