里面检测没有nan值、不识别问题,就只有训练的时候有问题。
下面正式开始解决问题,在train.py搜索amp把check_amp注释掉直接把amp赋值为False,如下图:
这样做之后在运行train.py发现训练时就不会有nan值了。
解决方法参照:http://t.csdn.cn/f8Vri
里面检测没有nan值、不识别问题,就只有训练的时候有问题。
下面正式开始解决问题,在train.py搜索amp把check_amp注释掉直接把amp赋值为False,如下图:
这样做之后在运行train.py发现训练时就不会有nan值了。
解决方法参照:http://t.csdn.cn/f8Vri