search

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras
MIT License
4.01k stars 412 forks source link

Change the default device to the model #145

Closed z-a-f closed 3 years ago

z-a-f commented 4 years ago

Change the default device to the one in the model