edit predit

2025-12-16 18:52:14 +01:00
parent 565c8e4855
commit b2d8ba1810
1 changed files with 1 additions and 1 deletions
@@ -50,7 +50,7 @@ model = cls(
            device=DEVICE,
        )

-model.load_state_dict(torch.load("LSTM_wu900_ps150_aw0.001.pt"))
+model.load_state_dict(torch.load("LSTM_wu900_ps150_aw0.001.pt", map_location=torch.device('cpu')))
 model.to(DEVICE)
 model.eval()
 with torch.no_grad():