Using to Caffe2 to create a model that uses dropout but getting an error related to dropout code

Question

I'm trying to create a model in Caffe2 that uses dropout. But I'm getting a model an error that refers to my code dropout code.

def someModel(model, data):

    conv1 = brew.conv(model, data, 'conv1', dim_in=1, dim_out=20, kernel=5)
    conv_relu_1 = model.net.Relu(conv1, 'relu1')

    conv2 = brew.conv(model, conv_relu_1, 'conv2', dim_in=1, dim_out=20, kernel=5)
    conv_relu_2 = model.net.Relu(conv2, 'relu2')

    pool1 = model.net.MaxPool(conv_relu_2, 'pool1', kernel=2, stride=2)
    drop1 = model.Dropout(pool1, 'drop1', ratio=0.5, is_test=0)
    #drop1 = model.Dropout(pool1, 'drop1', ratio=0.5)

    conv3 = brew.conv(model, drop1, 'conv3', dim_in=1, dim_out=50, kernel=3)
    conv_relu_3 = model.net.Relu(conv3, 'relu3')

    conv4 = brew.conv(model, conv_relu_3, 'conv4', dim_in=1, dim_out=20, kernel=5)
    conv_relu_4 = model.net.Relu(conv4, 'relu4')

    pool2 = model.net.MaxPool(conv_relu_4, 'pool1', kernel=2, stride=2)
    drop2 = model.Dropout(pool2, 'drop2', ratio=0.5)

    fc1 = brew.fc(model, drop2, 'fc1', dim_in=20 * 4 * 4, dim_out=50)
    fc_relu_1 = model.net.Relu(fc1, 'relu5')
    fc2 = brew.fc(model, fc_relu_1, 'fc2', dim_in=50 * 4 * 4, dim_out=10)

    pred = brew.fc(model, fc2, 'pred', 500, 10)
    softmax = model.net.Softmax(pred, 'softmax')
    return softmax
    return pred

Below is the error I'm getting.

Exception when creating gradient for [Dropout]:[enforce fail at operator_gradient.h:86] schema->Verify(def_). (GradientMaker) Operator def did not pass schema checking: input: "pool1" output: "drop2" name: "" type: "Dropout" arg { name: "ratio" f: 0.5 } . Op: input: "pool1" output: "drop2" name: "" type: "Dropout" arg {name: "ratio" f: 0.5}

Answer 1

Define dropout layer as

dropout1 = brew.dropout(model,pool1, 'dropout1', ratio=0.5, is_test=0)