Save torchsummary

1

I know how to print out the model by

torchsummary.summary()

,So I want to know is there any way to save the text into a txt file that I don’t know how to save it? Thank you

2

Assuming you are using this method from torchsummary you could call:

result, params_info = summary_string(
        model, input_size, batch_size, device, dtypes)

and save the result into a file instead of printing it out.

1 Like
3

Thank you very much!