wassname/seq2seq-time
1
0
Fork 0
mirror of https://github.com/wassname/seq2seq-time.git synced 2026-09-09 11:34:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

lots of models, inc NP

Browse Source
This commit is contained in:
wassname
2020-10-19 20:51:23 +08:00
parent e986751e41
commit 7b6c729db5
12 changed files with 3554 additions and 1623 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seq2seq_time/models/baseline.py
+1 -1
View File
@@ -12,4 +12,4 @@ class BaselineLast(nn.Module):
B, S, F = future_x.shape
mean = past_y[:, -1:].repeat(1, S, 1)
std = (self.std * 1.0).repeat(1, S, 1)
return torch.distributions.Normal(mean, std)
return torch.distributions.Normal(mean, std), {}