eb73f59f2e132eee98d6cbf548b5a08b59607537,torchbearer/callbacks/cutout.py,CutMix,__init__,#CutMix#Any#Any#Any#,135

Before Change

    def __init__(self, alpha, classes=-1, seed=None):
        super(CutMix, self).__init__()
        self.classes = classes
        torch.manual_seed(seed)
        self.dist = Beta(torch.tensor([alpha]), torch.tensor([alpha]))

    def _to_one_hot(self, target):
        if target.dim() == 1:

After Change

    def __init__(self, alpha, classes=-1, seed=None):
        super(CutMix, self).__init__()
        self.classes = classes
        if seed is not None:
            torch.manual_seed(seed)
        self.dist = Beta(torch.tensor([float(alpha)]), torch.tensor([float(alpha)]))

    def _to_one_hot(self, target):
        if target.dim() == 1:

Instances