ac368987658176719fab4039d5d0e04942eead71,python/mxnet_benchmarks/models/alexnet.py,AlexNet,__init__,#AlexNet#Any#,30

Before Change

    def __init__(self, params):
        Model.check_parameters(
            params,
            {"name": "AlexNet", "input_shape":(3, 227, 227), "num_classes": 1000,
             "phase": "training",
             "dtype": "float32"}
        )

After Change

        flatten = mx.symbol.Flatten(data=pool5)
        fc6 = mx.symbol.FullyConnected(name="fc6", data=flatten, num_hidden=4096)
        relu6 = layers.Activation(name="relu6", data=fc6, act_type="relu")
        drop6 = layers.Dropout(name="drop6", data=relu6, p=0.5)

        fc7 = mx.symbol.FullyConnected(name="fc7", data=drop6, num_hidden=4096)
        relu7 = layers.Activation(name="relu7", data=fc7, act_type="relu")
        drop7 = layers.Dropout(name="drop7", data=relu7, p=0.5)

Instances