Create a StochasticGradient class, using the given Module and Criterion. The class contains several parameters you might want to set after initialization.