How to save tw.draw_model to image file

[jackieyung]

I can only run my model on server, so I can't see the model graph directly from the example code: tw.draw_model. So, anyone can tell me how to save the graph data to image file? Thanks.

[Yuuchuin]

The result of tw.draw_model is a object of DotWrapper Calss. You can save tw.draw_model to image file as follow:

img = tw.draw_model(model, [1, 3, 224, 224]) # list cantains the size of input tensor img.save(filename)

[yaochx]

I find the method help info by help(tw.draw_model) draw_model(model, input_shape=None, orientation='TB', png_filename=None) after running this code tw.draw_model(model, [1, 3, 224, 224], 'model.png') nothing was saved

using img.save, the png resolution is too low to see the model graph