wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-09-13 13:02:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove comments and cleanup

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-14 11:55:33 +01:00
parent 6979042497
commit a1fe683254
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/tempflow/tempflow_network.py
+1 -5
View File
@@ -267,8 +267,6 @@ class TempFlowTrainingNetwork(nn.Module):
past_observed_values[:, -self.context_length :, ...],
)
# import pdb; pdb.set_trace()
outputs, states, lags_scaled, inputs = self.unroll(
lags=lags,
scale=scale,
@@ -363,7 +361,7 @@ class TempFlowTrainingNetwork(nn.Module):
# unroll the decoder in "training mode", i.e. by providing future data
# as well
rnn_outputs, _, scale, _, inputs = self.unroll_encoder(
rnn_outputs, _, scale, _, _ = self.unroll_encoder(
past_time_feat=past_time_feat,
past_target_cdf=past_target_cdf,
past_observed_values=past_observed_values,
@@ -387,8 +385,6 @@ class TempFlowTrainingNetwork(nn.Module):
# (batch_size, subseq_length, 1)
likelihoods = -distr.log_prob(target).unsqueeze(-1)
# import pdb; pdb.set_trace()
# assert_shape(likelihoods, (-1, seq_len, 1))
past_observed_values = torch.min(