Skip to content

Commit da94c16

Browse files
committed
Update gluontools.py
1 parent ba279ec commit da94c16

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lab/gluontools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def train(ctx,
8686
print(epoch_s + time_s)
8787
train_history.update([1 - train_acc, 1 - val_acc]) # 更新图像的纵轴
8888
train_history.plot(save_path=f'{resultDir}/{modelName}_history.png') # 实时更新图像
89-
if abs(train_acc-val_acc)>.3:
89+
if abs(train_acc-val_acc)>.25: # 严重过拟合
9090
break
9191
if val_acc > best_val_score: # 保存比较好的模型
9292
best_val_score = val_acc

0 commit comments

Comments
 (0)