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...