1 parent ba279ec commit da94c16Copy full SHA for da94c16
lab/gluontools.py
@@ -86,7 +86,7 @@ def train(ctx,
86
print(epoch_s + time_s)
87
train_history.update([1 - train_acc, 1 - val_acc]) # 更新图像的纵轴
88
train_history.plot(save_path=f'{resultDir}/{modelName}_history.png') # 实时更新图像
89
- if abs(train_acc-val_acc)>.3:
+ if abs(train_acc-val_acc)>.25: # 严重过拟合
90
break
91
if val_acc > best_val_score: # 保存比较好的模型
92
best_val_score = val_acc
0 commit comments