I wangt to use amp for mixed-precision training. But when i set amp.autocast is true:

image889×427 42.3 KB

I met the problem as follows:

image950×89 7.54 KB

What puzzles me is i can train the model normally, when i set the amp.autocast is false. So how can i to solve the problem?