RuntimeError: Placeholder storage has not been allocated on MPS device!

[Fyu]

Ha… the issue is the missing device=device in nn.Linear()