|
|
|
@ -3701,6 +3701,23 @@ class TestLayerParameterTrainableSet(unittest.TestCase):
|
|
|
|
|
self.assertFalse(net.weight.trainable)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestLayerTrainingAttribute(unittest.TestCase):
|
|
|
|
|
def test_set_train_eval_in_dynamic_mode(self):
|
|
|
|
|
with fluid.dygraph.guard():
|
|
|
|
|
net = paddle.nn.Dropout()
|
|
|
|
|
net.train()
|
|
|
|
|
self.assertTrue(net.training)
|
|
|
|
|
net.eval()
|
|
|
|
|
self.assertFalse(net.training)
|
|
|
|
|
|
|
|
|
|
def test_set_train_eval_in_static_mode(self):
|
|
|
|
|
net = paddle.nn.Dropout()
|
|
|
|
|
net.train()
|
|
|
|
|
self.assertTrue(net.training)
|
|
|
|
|
net.eval()
|
|
|
|
|
self.assertFalse(net.training)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
paddle.enable_static()
|
|
|
|
|
unittest.main()
|
|
|
|
|