mindspore-lab / mindcv

A toolbox of vision models and algorithms based on MindSpore
https://mindspore-lab.github.io/mindcv/
Apache License 2.0
230 stars 139 forks source link

fix deeplabv3 loss issue via SATURATION_MODE #775

Closed PingqiLi closed 4 months ago

PingqiLi commented 4 months ago

Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:

Motivation

Fix deeplabv3 loss getting nan issue by setting the SATURATION MODE environment variable. 通过设置SATURATION MODE环境变量修复deeplabv3 loss变为nan的问题。

Test Plan

(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)

Related Issues and PRs

(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)