zheng-da commented on issue #8302: Refactor operators & MKLDNN URL: https://github.com/apache/incubator-mxnet/pull/8302#issuecomment-359960347 Does it run the model in the inference mode? On Tue, Jan 23, 2018 at 1:25 PM, Sheng Zha <notificati...@github.com> wrote: > *@szha* commented on this pull request. > ------------------------------ > > In tests/python/gpu/test_gluon_model_zoo.py > <https://github.com/apache/incubator-mxnet/pull/8302#discussion_r163381644> > : > > > @@ -33,6 +33,8 @@ def download_data(): > 'http://data.mxnet.io/data/val-5k-256.rec', VAL_DATA) > > def test_models(): > + # We use network models without dropout for testing. > + # TODO(zhengda) mobilenet can't pass this test even without MKLDNN. > all_models = ['resnet18_v1', 'densenet121', 'mobilenet1.0'] > > You can disable dropout by > > with autograd.record(train_mode=False): > ... > > Doc at https://mxnet.incubator.apache.org/versions/master/ > api/python/autograd/autograd.html?highlight=autograd. > record#mxnet.autograd.record > > ? > You are receiving this because you were mentioned. > Reply to this email directly, view it on GitHub > <https://github.com/apache/incubator-mxnet/pull/8302#pullrequestreview-90986856>, > or mute the thread > <https://github.com/notifications/unsubscribe-auth/AAETUbk4rb1Zt5xgsnSXBWJBz0t9Wt61ks5tNk5cgaJpZM4P7Qml> > . >
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
