Maintaining Project Compatibility for Python 3.9 Users Without Upgrade Requirements.

Sole usage of Python 3.10's match-case in the project hinders quick-start for beginners; consider replacing with if-else for improved accessibility.
This commit is contained in:
cuba3 2024-06-25 15:24:46 +08:00
parent a30b19dd55
commit 9e60cdbc3f

View File

@ -20,7 +20,6 @@ def get_param(model) -> torch.nn.Parameter:
def float64(t: torch.Tensor):
"""return torch.float64 if device is not mps or xpu, else return torch.float32"""
match t.device.type:
case 'mps', 'xpu':
return torch.float32
if t.device.type in ['mps', 'xpu']:
return torch.float32
return torch.float64