wassname/attentive-neural-processes
1
0
Fork 0
mirror of https://github.com/wassname/attentive-neural-processes.git synced 2026-06-27 19:29:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

misc

Browse Source
This commit is contained in:
wassname
2019-11-03 10:05:48 +08:00
parent 24485b65d0
commit be042a91bb
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
readme.md
+1 -1
View File
@@ -19,7 +19,7 @@ This repository also includes a pytorch implementation that has been tweaked to
- Time features: time of day, day of week, month of year, etc
- Bank holidays
- Position in sequence: days since start of window
- Target is: mean power usage on block
- Target is: mean power usage on block 0
## Example outputs
src/models/modules.py
+1 -1
View File
@@ -253,7 +253,7 @@ class Decoder(nn.Module):
x = self._target_transform(target_x)
# concatenate target_x and representation
if r is not None:
if self.use_deterministic_path:
z = torch.cat([r, z], dim=-1)
representation = torch.cat([z, x], dim=-1)