CPU 单核和多核都能够稳定复现。随机数种子已经设定
np.random.seed(0) random.seed(0) train_prog.random_seed = args.random_seed startup_prog.random_seed = args.random_seed
a11xaf1n1#
@aprilvkuo 你是GPU计算结果有diff还是CPU?
z4iuyo4d2#
CPU多次运行结果无diff, GPU的计算结果多次运行结果有diff
mspsb9vt3#
加上这个试试 FLAGS_cudnn_deterministic=True
kt06eoxx4#
@DDDivano你好, 加上这个之后,效果还是有diff
4条答案
按热度按时间a11xaf1n1#
@aprilvkuo 你是GPU计算结果有diff还是CPU?
z4iuyo4d2#
CPU多次运行结果无diff, GPU的计算结果多次运行结果有diff
mspsb9vt3#
加上这个试试 FLAGS_cudnn_deterministic=True
kt06eoxx4#
@DDDivano
你好, 加上这个之后,效果还是有diff