Skip to content

Commit 4e9d287

Browse files
[fsdp, trainer] fix: save config parameters to wandb in SFT (#2884)
1 parent 0da8eb6 commit 4e9d287

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

verl/trainer/fsdp_sft_trainer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
import hydra
3131
import torch
3232
import torch.distributed
33-
from omegaconf import DictConfig
33+
from omegaconf import DictConfig, OmegaConf
3434
from peft import LoraConfig, TaskType, get_peft_model
3535
from tensordict import TensorDict
3636
from torch import nn, optim
@@ -688,6 +688,7 @@ def fit(self):
688688
project_name=self.config.trainer.project_name,
689689
experiment_name=self.config.trainer.experiment_name,
690690
default_backend=self.config.trainer.logger,
691+
config=OmegaConf.to_container(self.config, resolve=True),
691692
)
692693

693694
global_step = self.resume_global_step # Start from resumed step

0 commit comments

Comments
 (0)