def _do_after_cal_1(self, x, test):
if self.noise:
x = add_noise(x, test=test)
if self.dropout:
x = F.dropout(x, train=not test)
return x
微信
公众号
扫码关注公众号