search

zhougr1993 / DeepInterestNetwork

1.62k stars 558 forks source link

关于代码中的bn实现部分的疑问 #83

Open ghost opened 3 years ago

ghost commented 3 years ago

hist_i = tf.layers.batch_normalization(inputs = hist_i) 如上代码，请问下为什么bn没有设置training参数呢？ @zhougr1993

GuoshenLi commented 3 years ago

我也觉得肯定有问题的 要有training参数 并且要 updat_ops操作 可能在新版dien有解决吧

Waydrow commented 3 years ago

我也觉得肯定有问题的 要有training参数 并且要 updat_ops操作 可能在新版dien有解决吧

DIEN源码也是这么写的，没有加training参数。不过我测试了一下，发现加上training参数效果反而变差了，有点迷惑...