Created by: zhaoyang-star
[背景] 在conv-winograd.cc
中,在使用weights_
前如果不对其初始化为零,部分模型(如VIS图像修复模型)计算结果会出现偶发性错误。错误原因为在执行 winograd conv 时,会读取到weights_
中未被赋值的值(一般是个极大/极小或NAN),进而导致该层输出错误。
[解决方法] 在conv-winograd.cc
中,当使用weights_
时,提前初始化为零。已在VIS图像修复模型上验证通过。
Created by: zhaoyang-star
[背景] 在conv-winograd.cc
中,在使用weights_
前如果不对其初始化为零,部分模型(如VIS图像修复模型)计算结果会出现偶发性错误。错误原因为在执行 winograd conv 时,会读取到weights_
中未被赋值的值(一般是个极大/极小或NAN),进而导致该层输出错误。
[解决方法] 在conv-winograd.cc
中,当使用weights_
时,提前初始化为零。已在VIS图像修复模型上验证通过。