1. 修改了一些bug

2. 做了一些调整,比如启动脚本,支持了tenmsorboard
This commit is contained in:
2026-01-29 15:51:45 +08:00
parent e9e0e21e47
commit e494c4ce50
20 changed files with 995 additions and 287 deletions

View File

@@ -1,6 +1,6 @@
#!/bin/bash
# YG_FT_Base 统一启动脚本
# 同时启动后端 API 服务Web 静态服务器
# 同时启动后端 API 服务Web 静态服务器和 TensorBoard
# 使用方法: bash start_all.sh
# 自动修复脚本换行符
@@ -34,6 +34,7 @@ fi
echo "📦 端口配置:"
echo " - 后端 API: $API_PORT"
echo " - Web 服务器: $WEB_PORT"
echo " - TensorBoard: 6006"
echo ""
# 检查端口是否已被占用
@@ -70,6 +71,36 @@ start_api() {
echo "$API_PID" > /tmp/ygft_api.pid
}
# 启动 TensorBoard 服务
start_tensorboard() {
echo ""
echo "🚀 启动 TensorBoard 服务..."
# 检查端口
if ! check_port 6006; then
echo "⚠️ 端口 6006 已被占用TensorBoard 可能已在运行"
return 0
fi
# 确保日志目录存在
LOG_DIR="/app/base/saves"
if [ ! -d "$LOG_DIR" ]; then
LOG_DIR="$SCRIPT_DIR/saves"
fi
if [ ! -d "$LOG_DIR" ]; then
echo "⚠️ 日志目录不存在,跳过 TensorBoard 启动"
return 0
fi
# 启动 TensorBoard后台运行
nohup tensorboard --logdir "$LOG_DIR" --port 6006 --bind_all > "$LOG_DIR/tensorboard.log" 2>&1 &
TB_PID=$!
echo "✅ TensorBoard 服务已启动 (PID: $TB_PID, 端口: 6006)"
echo "$TB_PID" > /tmp/ygft_tensorboard.pid
echo "📊 TensorBoard 访问地址: http://localhost:6006"
}
# 启动 Web 静态服务器
start_web() {
echo ""
@@ -105,9 +136,16 @@ stop_all() {
echo "✅ Web 服务已停止"
fi
if [ -f /tmp/ygft_tensorboard.pid ]; then
kill $(cat /tmp/ygft_tensorboard.pid) 2>/dev/null
rm /tmp/ygft_tensorboard.pid
echo "✅ TensorBoard 服务已停止"
fi
# 清理可能残留的进程
pkill -f "src/main.py" 2>/dev/null
pkill -f "http.server $WEB_PORT" 2>/dev/null
pkill -f "tensorboard.*6006" 2>/dev/null
}
# 显示状态
@@ -128,16 +166,24 @@ status() {
echo "❌ Web 服务: 未运行"
fi
if [ -f /tmp/ygft_tensorboard.pid ] && kill -0 $(cat /tmp/ygft_tensorboard.pid) 2>/dev/null; then
echo "✅ TensorBoard: 运行中 (PID: $(cat /tmp/ygft_tensorboard.pid), 端口: 6006)"
else
echo "❌ TensorBoard: 未运行"
fi
echo ""
echo "🌐 访问地址:"
echo " - 后端 API: http://localhost:$API_PORT"
echo " - Web 页面: http://localhost:$WEB_PORT/pages/main.html"
echo " - TensorBoard: http://localhost:6006"
}
# 主菜单
case "$1" in
start)
start_api
start_tensorboard
start_web
echo ""
echo "===================================="
@@ -153,6 +199,7 @@ case "$1" in
stop_all
sleep 1
start_api
start_tensorboard
start_web
echo ""
echo "===================================="