diff --git a/perslay/perslay.py b/perslay/perslay.py index 520164b3..acae4048 100644 --- a/perslay/perslay.py +++ b/perslay/perslay.py @@ -158,7 +158,7 @@ def __init__(self, name, diagdim, perslay_parameters, rho): elif layer == "RationalHat": LMinit, LRinit = plp["lmean_init"], plp["lr_init"] LMiv = LMinit if not callable(LMinit) else LMinit([self.diagdim, plp["lnum"]]) - LRiv = LRinit if not callable(LRinit) else LVinit([1]) + LRiv = LRinit if not callable(LRinit) else LRinit([1]) # Fixed: was LVinit([1]) causing UnboundLocalError LM = tf.Variable(name=Lname+"-M", initial_value=LMiv, trainable=Ltrain) LR = tf.Variable(name=Lname+"-R", initial_value=LRiv, trainable=Ltrain) self.vars[nf].append([LM, LR])