"tests/models/openai/test_tokenization_openai.py" did not exist on "e75c3f70aa5122dbdd06cc693a58df837e422721"
Unverified Commit 54ba8608 authored by Bharat Ramanathan's avatar Bharat Ramanathan Committed by GitHub
Browse files

fix(integrations): store serialized `TrainingArgs` to `wandb.config` without sanitization. (#25035)



fix: store training args to wandb config without sanitization.

Allows resuming runs by reusing the wandb config.
Co-authored-by: default avatarBharat Ramanathan <ramanathan.parameshwaran@gohuddl.com>
parent 0906d212
...@@ -716,7 +716,7 @@ class WandbCallback(TrainerCallback): ...@@ -716,7 +716,7 @@ class WandbCallback(TrainerCallback):
logger.info( logger.info(
'Automatic Weights & Biases logging enabled, to disable set os.environ["WANDB_DISABLED"] = "true"' 'Automatic Weights & Biases logging enabled, to disable set os.environ["WANDB_DISABLED"] = "true"'
) )
combined_dict = {**args.to_sanitized_dict()} combined_dict = {**args.to_dict()}
if hasattr(model, "config") and model.config is not None: if hasattr(model, "config") and model.config is not None:
model_config = model.config.to_dict() model_config = model.config.to_dict()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment