5e37bb3ea1bed3f329ff825eaa6c7b9ed92edb15,sentiment_discovery/reparameterization/reparameterization.py,Reparameterization,__call__,#Reparameterization#Any#Any#,134

Before Change

		sys.stdout.flush()
		_w = getattr(self.module, self.name)
		if not self.evaluated or _w is None:
			setattr(self.module, self.name, self.compute_weight(self.module))
			self.evaluated = True

	def backward_hook(self, module, grad_input, grad_output):

After Change

	def __call__(self, module, inputs):
		callable hook for forward pass
		module2use, name2use = Reparameterization.get_module_and_name(module, self.name)
		_w = getattr(module2use, name2use)
		if not self.evaluated or _w is None:
			setattr(module2use, name2use, self.compute_weight(module2use, name2use))

Instances