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

fix typo

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 14:47:07 +01:00
parent 6ffa46d0f5
commit 054104019c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/tempflow/tempflow_network.py
+1 -1
View File
@@ -65,7 +65,7 @@ class TempFlowTrainingNetwork(nn.Module):
self.dequantize = dequantize
self.distr_output = FlowOutput(
self.flow, input_size=input_size, cond_size=conditioning_length
self.flow, input_size=target_dim, cond_size=conditioning_length
)
self.proj_dist_args = self.distr_output.get_args_proj(num_cells)