actually expand

Tractables · Dec 25, 2024 · bac98f6 · bac98f6
1 parent b1f8684
commit bac98f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qc/benchmarks/benchmarks.jl b/qc/benchmarks/benchmarks.jl
@@ -451,11 +451,11 @@ function produce_loss(rs::RunState, m::FeatureSpecEntropyLossMgr, epoch::Integer
                 num_meeting += 1
 
                 lpr_eq = LogPr(prob_equals(m.generation.value, sample))
+                lpr_eq = Dice.expand_logprs(l, lpr_eq)
                 empirical_feature_logpr = log(feature_counts[m.p.feature(sample)]/length(samples))
                 if m.p.train_feature
                     [lpr_eq * empirical_feature_logpr, empirical_feature_logpr]
                 else
-                    lpr_eq = Dice.expand_logprs(l, lpr_eq)
                     [lpr_eq * compute(a, lpr_eq), empirical_feature_logpr]
                 end
             else