[Feature Request] save symbol as well in gluon

[ShownX]

I start to use gluon but I found it cannot save the model into the symbol file. Is it possible to implement this function?

[piiswrong]

Only hybridblock can be serialized to symbol. Normal blocks can only be pickled

[ShownX]

how to serialize hybridblock?

[szha]

you can call the .save_params() function on it.

[ShownX]

Yes. I tried .save_params() function, it only saves the params. I cannot find the symbol file :(.
I checked all layers. All of them are inherited from hybridblock.

[szha]

@ShownX ah sorry for missing that. To get the symbol out of a block, just feed the block with a symbol of the intended put variable. Something like block(sym.Variable('data')) and then you can use symbol.save.

[ShownX]

Suppose I wrote the following code:

net = gluon.nn.HybridSwquential()
net.add(gluon.nn.Dense(512))

Is it possible to save net definition to a symbol file? I cannot find the API like you said above. What I only find is net.save_params() to save weights.

[szha]

You can call net(mx.sym.Variable('data')) and the returned value is a symbol, with which you can save with the symbol API.

[ShownX]

I see. Thanks.