zhougr1993 / DeepInterestNetwork

1.61k stars 559 forks source link

fix bn not share bugs for din #70

Closed liuhu-bigeye closed 4 years ago

liuhu-bigeye commented 4 years ago

See: https://github.com/zhougr1993/DeepInterestNetwork/issues/63

The batch_norm params for hist_j should be shared with hist_i, otherwise it will be never trained and using hist_j in eval phase is equivalent to exposing the label.