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: Bharat Ramanathan <ramanathan.parameshwaran@gohuddl.com>
This commit is contained in:
Bharat Ramanathan 2023-07-24 18:12:39 +05:30 committed by GitHub
parent 0906d21203
commit 54ba8608d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -716,7 +716,7 @@ class WandbCallback(TrainerCallback):
logger.info(
'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:
model_config = model.config.to_dict()