Skip to content

Commit 9d73b42

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Add label for phase I-b generation output samples.
1 parent d4851fe commit 9d73b42

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1426,6 +1426,8 @@ def create_dataset(raw_text_samples, tokenizer, sample_expansion_batch_size=10)
14261426
result_phase_i_b = float(phase_i_b_history['perplexity'].min())
14271427
mlflow.log_metric("phase_i_b-perplexity", result_phase_i_b, step=trial_number)
14281428

1429+
print("########### Phase I-b Model Checkpoint Generation Samples: ")
1430+
14291431
# Text samples after Phase I-b training
14301432
counter = 0
14311433
for sample in prompt_samples:

0 commit comments

Comments
 (0)