Merge branch 'smaller-weight-c' of github.com:asappresearch/sru into …

…smaller-weight-c # Conflicts: # sru/modules.py
asappresearch · Dec 17, 2020 · 7c27bea · 7c27bea
2 parents f4e1687 + 2e67644
commit 7c27bea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sru/modules.py b/sru/modules.py
@@ -22,7 +22,7 @@ class SRUCell(nn.Module):
                      'dropout', 'bidirectional', 'has_skip_term', 'highway_bias',
                      'v1', 'rescale', 'activation_type', 'activation', 'custom_m',
                      'projection_size', 'num_matrices', 'layer_norm', 'weight_proj',
-                     'scale_x']
+                     'scale_x', 'normalize_after', 'weight_c_init',]
 
     scale_x: Tensor
     weight_proj: Optional[Tensor]