docs: 更新实验验证章节并添加相关图表
更新了实验验证章节的内容,包括硬件配置、LoRA微调参数、计算资源占用和训练指标变化的详细描述。新增了三张图表(monitor.png、training_metrics.png、after_train.png)以直观展示训练过程中的系统资源占用、训练指标变化和微调后的效果。这些修改增强了实验部分的完整性和可读性。
This commit is contained in:
34
paper/object/plot_training_data.py
Normal file
34
paper/object/plot_training_data.py
Normal file
@@ -0,0 +1,34 @@
|
||||
import pandas as pd
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
# 读取CSV文件
|
||||
data = pd.read_csv('training_data.csv')
|
||||
|
||||
# 创建图表
|
||||
plt.figure(figsize=(12, 9))
|
||||
|
||||
# 绘制梯度范数变化曲线
|
||||
plt.subplot(3, 1, 1) # 修改为 3行1列的第1个
|
||||
plt.plot(data['Step'], data['grad_norm'], label='Gradient Norm')
|
||||
plt.xlabel('Step')
|
||||
plt.ylabel('Gradient Norm')
|
||||
plt.legend()
|
||||
|
||||
# 绘制损失值变化曲线
|
||||
plt.subplot(3, 1, 2) # 修改为 3行1列的第2个
|
||||
plt.plot(data['Step'], data['loss'], label='Loss', color='orange')
|
||||
plt.xlabel('Step')
|
||||
plt.ylabel('Loss')
|
||||
plt.legend()
|
||||
|
||||
# 绘制学习率变化曲线
|
||||
plt.subplot(3, 1, 3) # 修改为 3行1列的第3个
|
||||
plt.plot(data['Step'], data['learning_rate'], label='Learning Rate', color='green')
|
||||
plt.xlabel('Step')
|
||||
plt.ylabel('Learning Rate')
|
||||
plt.legend()
|
||||
|
||||
# 调整布局并保存图片
|
||||
plt.tight_layout()
|
||||
plt.savefig('training_metrics.png')
|
||||
plt.show()
|
1001
paper/object/training_data.csv
Normal file
1001
paper/object/training_data.csv
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user