Update generative-proof-of-concept-CPU-preprocessing-in-memory.py

Fix broken refernce to variable name changed; Add phase_i_b_epochs to logged params.

Author: david-thrower

generative-proof-of-concept-CPU-preprocessing-in-memory.py

@@ -173,6 +173,7 @@ def objective(trial: optuna.Trial) -> float:
               "num_lateral_connection_tries_per_unit": num_lateral_connection_tries_per_unit,
               "learning_rate":learning_rate,
               "epochs":epochs,
+              "phase_i_b_epochs": phase_i_b_epochs,
               "batch_size":batch_size,
               "gradient_accumulation_steps":gradient_accumulation_steps,
               "minimum_levels":minimum_levels,
@@ -1257,7 +1258,7 @@ def test_text(test_prompt: str, max_new_tokens: int, sample_number: int, result:
                    test_prompt=sample,
                    max_new_tokens=MAX_NEW_TOKENS,
                    sample_number=counter,
-                   result=result,
+                   result=phase_i_a_result,
                    result_cutoff=RESULT_CUTOFF,
                    trial_id=trial_number,
                    test_sample_number=counter,