sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

4.01k stars 412 forks source link

add checks for atomic values in output #111

Open wyc1997 opened 4 years ago

wyc1997 commented 4 years ago

110

A possible fix to this problem.