Martin
Results
1
issues of
Martin
**Issue Description** Specifying device='mps' results in the following error. ```Python AssertionError: Input device is not valid, please specify 'cuda' or 'cpu' ``` **To Reproduce** ```Python from torchsummary import summary import...