Closed zhisays closed 4 years ago
is_training 是示例函数的bool类型参数而非函数
我们可以通过参数is_training函数来判断运行模式为训练还是测试,并只需在训练模式下使用丢弃法。
drop_prob1, drop_prob2 = 0.2, 0.5 def net(X, is_training=True): X = X.view(-1, num_inputs)
已更正,感谢指出。
is_training 是示例函数的bool类型参数而非函数