pass in device and dtypes explicitly for #102, among others - Githubissues

sksq96 / pytorch-summary

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

MIT License

4.01k stars 412 forks source link

pass in device and dtypes explicitly for #102, among others #107

Closed Naireen closed 4 years ago