7b460340070c810de6d0f6288f6c8900c93e6cfa,torchsample/modules/super_module.py,SuperModule,set_optimizer,#SuperModule#Any#,49

Before Change

        self._loss = loss

    def set_optimizer(self, optimizer, **kwargs):
        self._optimizer = optimizer(self.parameters(), **kwargs)

    def set_regularizers(self, regularizers):
        self._regularizers = regularizers

After Change

            parameters = kwargs["parameters"]
        else:
            parameters = self.parameters()
        self._optimizer = optimizer(parameters, **kwargs)

    def set_regularizers(self, regularizers):
        self._regularizers = regularizers

Instances