liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-17 09:24:34 +08:00
Code Activity

Add torch.no_grad for evaluation

Browse Source
This commit is contained in:
D-X-Y
2021-03-12 02:46:04 +00:00
parent 67fbdafe76
commit db59713d36
5 changed files with 35 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
qlib/contrib/model/pytorch_gru.py
View File

@@ -208,12 +208,13 @@ class GRU(Model):
feature = torch.from_numpy(x_values[indices[i : i + self.batch_size]]).float().to(self.device)
label = torch.from_numpy(y_values[indices[i : i + self.batch_size]]).float().to(self.device)
pred = self.gru_model(feature)
loss = self.loss_fn(pred, label)
losses.append(loss.item())
with torch.no_grad():
pred = self.gru_model(feature)
loss = self.loss_fn(pred, label)
losses.append(loss.item())
score = self.metric_fn(pred, label)
scores.append(score.item())
score = self.metric_fn(pred, label)
scores.append(score.item())
return np.mean(losses), np.mean(scores)