def test_dropout_training(net):
    """Test dropout `.training` is set by `.train()` on parent `nn.module`"""
    net.train()
    assert net._dropout.training == True
Test dropout `.training` is set by `.train()` on parent `nn.module`
tests/test_model.py