Thank you all very much for the help. It wasn’t that clear but I could figure it out. The way I did it:
        ...

        scale = scaler.get_scale()
        scaler.update()
        skip_lr_sched = (scale != scaler.get_scale())
        ...
        
        if not skip_lr_sched:
            lr_sched.step()
        ...