Question about the visualization function 'cls_padding' in demo.ipynb

[star-journey]

I appreciate your interesting work. In the cls_padding function, there are two lines that seem to be a bug:

    mask = mask / max(np.max(mask),cls_weight)
    cls_weight = cls_weight / max(np.max(mask),cls_weight)

In the first line, the variable mask is modified. This means that the value of max(np.max(mask), cls_weight) in the second line will be different. I believe this should be a normalization operation. Can you confirm if this is a bug or a feature?