Save network plot to file about hiddenlayer HOT 3 OPEN

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")

from hiddenlayer.

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

from hiddenlayer.

Thanks for this fantastic library.

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

from hiddenlayer.