diff --git a/HousePrice/src/train.py b/HousePrice/src/train.py
index 9ef6319..5f19afa 100644
--- a/HousePrice/src/train.py
+++ b/HousePrice/src/train.py
@@ -34,7 +34,7 @@ def loss(z, y):
 optim = optim.Adam(model.parameters(), lr=learning_rate)
 
 step = 0
-right_num = 0
+right_num = 0.
 for epoch in range(1, EPOCHS+1):
 
     print(f'================第{epoch}轮================')