首先需要在 yaml 里设置两个参数:

output_dir: /home/wangguisen/projects/LLaMA-Factory/weights/tensbox_demo

report_to: tensorboard
logging_dir: /home/wangguisen/projects/LLaMA-Factory/weights/tensbox_demo/runs

然后开始训练,在你的输出目录下会有 runs 文件夹。

安装 TensorBoard

proxychains4 pip install tensorboard

然后在服务上使用默认的6006端口正常启动TensorBoard:

tensorboard --logdir=/home/wangguisen/projects/LLaMA-Factory/weights/tensbox_demo/runs --port=6006 --host=localhost

或者本地终端:

ssh -L 16006:localhost:6006 your_username@server_ip

16006 是你本地机器上的端口号,可以是任意未占用的端口。
6006 是服务器上的 TensorBoard 端口号。

然后浏览器打开链接: http://localhost:16006,即可通过 SSH 隧道访问 TensorBoard。

ref:

https://github.com/hiyouga/LLaMA-Factory/issues/816

https://huggingface.co/docs/transformers/main_classes/trainer#transformers.TrainingArguments.logging_dir

https://datawhalechina.github.io/thorough-pytorch/%E7%AC%AC%E4%B8%83%E7%AB%A0/7.3%20%E4%BD%BF%E7%94%A8TensorBoard%E5%8F%AF%E8%A7%86%E5%8C%96%E8%AE%AD%E7%BB%83%E8%BF%87%E7%A8%8B.html#id8

Logo

免费领 200 小时云算力,进群参与显卡、AI PC 幸运抽奖

更多推荐