Save network plot to file

[erap129]

Hello,

I saw there is a possibility to save training stat plots into png files, but I couldn't find an option to save the actual NN architecture plot into a file, is this currently possible?

Thanks

[mdhasanai]

To do this, first, save the picture(output) in a variable

im = hl.build_graph(model_name, torch.zeros([64,3,7,7]))

(your size in place of torch.zeros([64,3,7,7])

Then save it giving the path with a name of the file (you want), also adding format of the file at the beginning

im.save(path="path_of_file/name_of_file" , format="jpg")

[ldbro0]

Is there a way you can change the figure size or resolution?

[daniel347x]

Thanks for this fantastic library.

It would be fabulous indeed if the above information could make it onto the home page. Thanks!