Commit Graph

2 Commits (57e6d05a43e4bdf4575e520f1a04c17e80fe58cc)

Author SHA1 Message Date
w-e-w 9c8075ba8e torch_utils.float64
return torch.float64 if device is not mps or xpu, else return torch.float32
2024-05-16 23:16:50 +09:00
Aarni Koskela 5768afc776 Add utility to inspect a model's parameters (to get dtype/device) 2023-12-31 13:22:43 +02:00