pytorch backward時に出てくるメッセージをoffにしたい

Question

        mainQN.eval()
        for i in trin_x:
            optimizer.zero_grad()
            output = mainQN.forward(inputs[i],"net_q")
            loss = criterion(output,targets[i])
            loss=loss*weights[i]
            with amp.scale_loss(loss, optimizer) as scaled_loss:
                scaled_loss.backward() # ここを変更
            optimizer.step()

出てくるメッセージ

Gradient overflow.  Skipping step, loss scaler 0 reducing loss scale to 6.162975822039155e-33

このメッセージをoffにしたいです　なぜか出ないときがたまにあるのですが
学習時　これが大量にでて困ってます　
表示しないようにする方法はありませんか？
情報が足りなければいってください

Comments

[Flan.]

治ったと思ったんですが　うまくいってませんでした　だから質問しなおしました

[kabayan55]

質問文に、前の質問のリンクを貼った上で、質問し直した理由も書いておくと良いと思います。こちらの欄だと気づかない方がいらっしゃるかと思います。

Answer 1

このメッセージをoffにしたいです　

おそらくできません。backwardの引数など確認しましたが、この部分のon, offをする引数は無さそうでした。

なぜか出ないときがたまにあるのですが

出てくるメッセージについて、検索したり意味を考えたりしてみましたか？ 翻訳をするというのも一つの手段です。
もし既にされていたら、「〇〇のキーワードで調べたけど情報が得られなかった」「〇〇と解釈したが正しいかわからない」などと質問に添えておくといいと思います。

なぜ出るときと出ない時があるか、メッセージの意味を考えてみるとわかると思います。

Gradient overflow. Skipping step, loss scaler 0 reducing loss scale to 6.162975822039155e-33

Gradient overflow、勾配が発散しています。つまり、メッセージはloss関数の勾配が発散しているときに出ています。

勾配発散問題については、以下の記事が参考になりそうでした。
勾配の更新に関する問題