ó ùµÈ[c@s¢dZddgZddlZddlmZddlmZdd lmZdd lm Z defd „ƒYZ e eƒej j e jƒd ƒd „ZdS(sAlexnet, implemented in Gluon.tAlexNettalexnetiÿÿÿÿNi(tcpui(t HybridBlock(tnn(tbasecBs#eZdZdd„Zd„ZRS(sËAlexNet model from the `"One weird trick..." `_ paper. Parameters ---------- classes : int, default 1000 Number of classes for the output layer. iècKs3tt|ƒj||jƒ tjddƒ|_|jjƒÎ|jjtjddddddd d d ƒƒ|jjtj d d dd ƒƒ|jjtjddddd d d ƒƒ|jjtj d d dd ƒƒ|jjtjddd ddd d ƒƒ|jjtjddd ddd d ƒƒ|jjtjddd ddd d ƒƒ|jjtj d d dd ƒƒ|jjtj ƒƒ|jjtj dd d ƒƒ|jjtj dƒƒ|jjtj dd d ƒƒ|jjtj dƒƒWdQXtj |ƒ|_ WdQXdS(Ntprefixti@t kernel_sizei tstridesitpaddingit activationtrelut pool_sizeiiÀii€iiigà?(tsuperRt__init__t name_scopeRtHybridSequentialtfeaturestaddtConv2Dt MaxPool2DtFlattentDensetDropouttoutput(tselftclassestkwargs((sd/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/gluon/model_zoo/vision/alexnet.pyR's. !"! "! ! ! "cCs"|j|ƒ}|j|ƒ}|S(N(RR(RtFtx((sd/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/gluon/model_zoo/vision/alexnet.pythybrid_forwardAs(t__name__t __module__t__doc__RR(((sd/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/gluon/model_zoo/vision/alexnet.pyRs tmodelscKsHt|}|rDddlm}|j|dd|ƒd|ƒn|S(s”AlexNet model from the `"One weird trick..." `_ paper. Parameters ---------- pretrained : bool, default False Whether to load the pretrained weights for model. ctx : Context, default CPU The context in which to load the pretrained weights. root : str, default $MXNET_HOME/models Location for keeping the model parameters. i(tget_model_fileRtroottctx(Rt model_storeR$tload_parameters(t pretrainedR&R%RtnetR$((sd/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/gluon/model_zoo/vision/alexnet.pyRGs "(R"t__all__tostcontextRtblockRRRRRtFalsetpathtjointdata_dirR(((sd/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/gluon/model_zoo/vision/alexnet.pyts  (