Minor tweaks and quick fix

sksq96 / pytorch-summary

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

MIT License

4.01k stars 412 forks source link

Open leriomaggio opened 4 years ago

leriomaggio commented 4 years ago

Removed unused import, PEP8 reformatting, and more robust checking of None for dtype

leriomaggio commented 4 years ago

Up @sksq96