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

get args takes in in_feature argument

Browse Source
This commit is contained in:
Kashif Rasul
2019-11-21 23:13:38 +01:00
parent eb67bf0b9a
commit e81d244832
3 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/modules/test_distribution_output.py
+1 -2
View File
@@ -27,8 +27,7 @@ def maximum_likelihood_estimate_sgd(distr_output: DistributionOutput,
init_biases: List[np.ndarray] = None,
num_epochs: int = 5,
learning_rate: float = 1e-2):
distr_output.in_features = 1
arg_proj = distr_output.get_args_proj()
arg_proj = distr_output.get_args_proj(in_features=1)
if init_biases is not None:
for param, bias in zip(arg_proj.proj, init_biases):