Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

embedding微调时,tensorboard可视化参数问题 #17

Closed
HuXiLiFeng opened this issue Jun 2, 2024 · 3 comments
Closed

embedding微调时,tensorboard可视化参数问题 #17

HuXiLiFeng opened this issue Jun 2, 2024 · 3 comments

Comments

@HuXiLiFeng
Copy link

https://github.com/NLPJCL/RAG-Retrieval/blob/master/rag_retrieval/train/embedding/train_embedding.py#L83
accelerator.init_trackers('embedding', config=args)
在该函数定义里,config是dict类型,但是这里的args是<class 'argparse.Namespace'>类型不符,报错:TypeError: hparam_dict and metric_dict should be dictionary.
按如下修改即可:
accelerator.init_trackers('embedding', config=vars(args))

@NLPJCL
Copy link
Owner

NLPJCL commented Jun 2, 2024

感谢提醒~
我这边默认用的是wandb来可视化,没有报过这个错误,您可以把执行的参数发下,我来试试。

@HuXiLiFeng
Copy link
Author

感谢提醒~ 我这边默认用的是wandb来可视化,没有报过这个错误,您可以把执行的参数发下,我来试试。

直接执行的embedding下的train_embedding.sh,只是换了下模型地址,log_with改成tensorboard。其他参数没动

@NLPJCL
Copy link
Owner

NLPJCL commented Jun 3, 2024

感谢提醒~ 我这边默认用的是wandb来可视化,没有报过这个错误,您可以把执行的参数发下,我来试试。

直接执行的embedding下的train_embedding.sh,只是换了下模型地址,log_with改成tensorboard。其他参数没动

多谢,我回头测试下。确实没尝试过更换为tensorboard。

@NLPJCL NLPJCL closed this as completed Jul 23, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants