saber icon indicating copy to clipboard operation
saber copied to clipboard
verified publisher icon BaderLab

Need nicely formatted string representation of a PyTorch model

Open JohnGiorgi opened this issue 6 years ago • 0 comments

Need to add a way to print out a nicely formatted representation of a PyTorch model. Wrap this under a summary() function in the BasePyTorchModel so that its usage is the same as the Keras models in Saber.

See here for inspiration.

JohnGiorgi avatar Jun 03 '19 13:06 JohnGiorgi