yuyijie1995 / dropblock_mxnet_bottom_implemention Goto Github PK

In [12]: x = mx.nd.ones(shape=(1, 1, 10, 10))

In [13]: out = mx.nd.GPUDropblock(x, p = 0.2,block_size=3, mode='always')

In [14]: out
Out[14]: 

[[[[1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]
   [1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024 1.00024
    1.00024 1.00024]]]]
<NDArray 1x1x10x10 @cpu(0)>

In [15]: x = mx.nd.ones(shape=(1, 1, 10, 10), ctx=mx.gpu(0))
[19:23:20] src/base.cc:84: Upgrade advisory: this mxnet has been built against cuDNN lib version 7401, which is older than the oldest version tested by CI (7600).  Set MX
NET_CUDNN_LIB_CHECKING=0 to quiet this warning.


In [16]: 

In [16]: x
Out[16]: 

[[[[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]]]]
<NDArray 1x1x10x10 @gpu(0)>

In [17]: out = mx.nd.GPUDropblock(x, p = 0.2,block_size=3, mode='always')

In [18]: out
Out[18]: 

[[[[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]]]]
<NDArray 1x1x10x10 @gpu(0)>

In [19]: with mx.autograd.record():
    ...:     out = mx.nd.GPUDropblock(x, p = 0.2,block_size=3, mode='always')
    ...:     

In [20]: 

In [20]: out
Out[20]: 

[[[[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
   [1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]]]]
<NDArray 1x1x10x10 @gpu(0)>