Skip to content

Commit bee4662

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Fix typo.
1 parent 165a64d commit bee4662

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -759,11 +759,13 @@ def create_dataset(raw_text_samples, tokenizer, sample_expansion_batch_size=50,
759759

760760
TOKENIZER_SAVE_PATH = f"tokenizer-tr-{trial}"
761761
tokenizer.save_pretrained(TOKENIZER_SAVE_PATH)
762-
print(f"Final model saved to {MODEL_SAVE_PATH}")
762+
print(f"Tokenizer saved to {TOKENIZER_SAVE_PATH}")
763763

764+
764765
MODEL_SAVE_PATH = f"final_phase_ib_model_tr_{trial}.keras"
765766
generator.save(MODEL_SAVE_PATH)
766-
print(f"Tokenizer saved to {TOKENIZER_SAVE_PATH}")
767+
print(f"Final model saved to {MODEL_SAVE_PATH}")
768+
767769

768770
# Return the final result to Optuna
769771
return result_phase_i_b

0 commit comments

Comments
 (0)