diff --git a/unet.py b/unet.py
index c03e023..12157e2 100644
--- a/unet.py
+++ b/unet.py
@@ -23,7 +23,7 @@ class Unet(object):
         #   训练好后logs文件夹下存在多个权值文件，选择验证集损失较低的即可。
         #   验证集损失较低不代表miou较高，仅代表该权值在验证集上泛化性能较好。
         #-------------------------------------------------------------------#
-        "model_path"        : 'model_data/unet_voc',
+        "model_path"        : 'model_data/unet_voc.h5',
         #----------------------------------------#
         #   所需要区分的类的个数+1
         #----------------------------------------#