def __init__(self, ): super(MultiCriterion, self).__init__() self.criterions = [] self.weights = torch.DoubleStorage()