首先需要在 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: