49e58fcd4fcbc3b596a1fced1f74b9a63464f0e8,parlai/core/torch_classifier_agent.py,TorchClassifierAgent,build_criterion,#TorchClassifierAgent#,156

Before Change

    def build_criterion(self):
        weight_tensor = torch.FloatTensor(self.class_weights)
        self.criterion = torch.nn.CrossEntropyLoss(weight_tensor)

    def share(self):
        Share model parameters.
        shared = super().share()

After Change

    def build_criterion(self):
        weight_tensor = torch.FloatTensor(self.class_weights)
        return torch.nn.CrossEntropyLoss(weight_tensor)

    def share(self):
        Share model parameters.
        shared = super().share()

Instances