liqian 3 years ago
parent
commit
0d6fa8d4e4
1 changed files with 1 additions and 1 deletions
  1. 1 1
      rov_train.py

+ 1 - 1
rov_train.py

@@ -220,6 +220,6 @@ if __name__ == '__main__':
 
     log_.info('rov model predict start...')
     predict_start = time.time()
-    predict()
+    predict_test()
     predict_end = time.time()
     log_.info('rov model predict end, execute time = {}ms'.format((predict_end - predict_start)*1000))