code: def reshape(self, data_shapes, label_shapes=None):
          
          super(SVRGModule, self).reshape(data_shapes, label_shapes=label_shapes)
          self._mod_aux.reshape(data_shapes, label_shapes=label_shapes)
docs: Reshapes both modules for new input shapes.
  
          Parameters
          ----------
          data_shapes : list of (str, tuple)
              Typically is ``data_iter.provide_data``.
          label_shapes : list of (str, tuple)
              Typically is ``data_iter.provide_label``.