lushizijizoude

I have soloved the problem. It's due to the function log(). model.py log_gate_score_1 = (gate_score+ 1e-45).log() # shape: (batch_size,) log_gate_score_2 = (1 - gate_score+ 1e-45).log() # shape: (batch_size,)