Commit 12a9dce2 authored by David Sussillo's avatar David Sussillo Committed by GitHub
Browse files

Merge pull request #2050 from djoshea/master

LFADS: Fixing alignment bias bug
parents 582bf927 41d700ef
...@@ -409,6 +409,11 @@ class LFADS(object): ...@@ -409,6 +409,11 @@ class LFADS(object):
dataset = datasets[name] dataset = datasets[name]
in_mat_cxf = dataset['alignment_matrix_cxf'].astype(np.float32) in_mat_cxf = dataset['alignment_matrix_cxf'].astype(np.float32)
if datasets and 'alignment_bias_c' in datasets[name].keys():
dataset = datasets[name]
align_bias_c = dataset['alignment_bias_c'].astype(np.float32)
align_bias_1xc = np.expand_dims(align_bias_c, axis=0)
out_mat_fxc = None out_mat_fxc = None
out_bias_1xc = None out_bias_1xc = None
if in_mat_cxf is not None: if in_mat_cxf is not None:
......
...@@ -112,7 +112,7 @@ def init_linear(in_size, out_size, do_bias=True, mat_init_value=None, ...@@ -112,7 +112,7 @@ def init_linear(in_size, out_size, do_bias=True, mat_init_value=None,
'Provided mat_init_value must have shape [%d, %d].'%(in_size, out_size)) 'Provided mat_init_value must have shape [%d, %d].'%(in_size, out_size))
if bias_init_value is not None and bias_init_value.shape != (1,out_size): if bias_init_value is not None and bias_init_value.shape != (1,out_size):
raise ValueError( raise ValueError(
'Provided bias_init_value must have shape [1,%d].'%(1,out_size)) 'Provided bias_init_value must have shape [1,%d].'%(out_size,))
if mat_init_value is None: if mat_init_value is None:
stddev = alpha/np.sqrt(float(in_size)) stddev = alpha/np.sqrt(float(in_size))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment