matteosal commented on issue #20109:
URL:
https://github.com/apache/incubator-mxnet/issues/20109#issuecomment-819665575
Another question: in v1.x it was possible to query executors for their
allocated memory:
```
import mxnet as mx
sym = mx.sym.Convolution(
mx.sym.Variable('in'),
mx.sym.Variable('w'),
num_filter=16,
kernel=(3, 3),
no_bias=True
)
sym = mx.sym.relu(sym)
args = {'in': mx.nd.ones([64, 3, 300, 300]), 'w': mx.nd.ones([16, 3, 3, 3])}
ex = sym.bind(mx.cpu(), args)
print(ex.debug_str())
```
```
Symbol Outputs:
output[0]=relu0(0)
Variable:in
Variable:w
--------------------
Op:Convolution, Name=convolution0
Inputs:
arg[0]=in(0) version=0
arg[1]=w(0) version=0
Attrs:
kernel=(3, 3)
no_bias=True
num_filter=16
--------------------
Op:relu, Name=relu0
Inputs:
arg[0]=convolution0(0)
Total 346 MB allocated
Total 11 TempSpace resource requested
```
Is there a way to get the equivalent of the line "Total 346 MB allocated"
with cached ops?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
