a72a5b05d8edcde1e1bbb808ed26d5c858b112d5,gluon/models/dpn.py,,_test,#,587

Before Change

        assert (model != dpn107 or weight_count == 86917800)
        assert (model != dpn131 or weight_count == 79254504)

        x = Variable(torch.randn(1, 3, 224, 224))
        y = net(x)
        assert (tuple(y.size()) == (1, 1000))

After Change

        assert (model != dpn131 or weight_count == 79254504)

        //net.hybridize()
        x = mx.nd.zeros((1, 3, 224, 224), ctx=ctx)
        y = net(x)
        assert (y.shape == (1, 1000))

Instances