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:
parent
0906d21203
commit
54ba8608d0
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue