def generate_sample(self, base_target, std_val=0.1):
    target = (base_target.float() / base_target.max())
    noise = (std_val * torch.rand(1, 1, 6, 5))
    return (target + noise)
def generate_sample(self, base_target, std_val=0.1):
    target = (base_target.float() / base_target.max())
    noise = (std_val * torch.rand(1, 1, 6, 5))
    return (target + noise)