task 3 p_mask h_mask 定义错误

[wxhheian]

local inference

p_mask, h_mask = (p != 1).long(), (h != 1).long() # TODO: 这里是不是写错了 p != 0
m_p, m_h = self.inference(p_, h_, p_mask, h_mask) # m_p = m_h = [256,50,2400]

p_mask, h_mask = (p != 0).long(), (h != 0).long() 因为填充值默认是0

[PandaPandaChen]

i think so