diff --git a/train.py b/train.py
index 50f093ea9fddb5f3bacc2c6e62c0d8f064f68fb0..ccaeab99610f1bea46fed4479461f424aed90798 100644
--- a/train.py
+++ b/train.py
@@ -92,4 +92,4 @@ if __name__ == "__main__":
             print('iter:' + str(iteration) + '/' + str(epoch_size) + ' || Loc_Loss: %.4f || Conf_Loss: %.4f ||' % (loc_loss/(iteration+1),conf_loss/(iteration+1)), end=' ')
 
         print('Saving state, iter:', str(epoch+1))
-        torch.save(model.state_dict(), 'logs/Epoch%d-Loc%.4f-Conf%.4f'%((epoch+1),loc_loss/(iteration+1),conf_loss/(iteration+1)))
+        torch.save(model.state_dict(), 'logs/Epoch%d-Loc%.4f-Conf%.4f.pth'%((epoch+1),loc_loss/(iteration+1),conf_loss/(iteration+1)))