val accuracy cal fix

22feea52 · meijieru · 4c8e7abf · 22feea52
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

crnn_main.py crnn_main.py +2 -1

未找到文件。
--- a/crnn_main.py
+++ b/crnn_main.py
@@ -137,7 +137,8 @@ def val(net, dataset, criterion, max_iter=100):
    n_correct = 0
    loss_avg = utils.averager()

-    for i in range(min(max_iter, len(data_loader))):
+    max_iter = min(max_iter, len(data_loader))
+    for i in range(max_iter):
        data = val_iter.next()
        i += 1
        cpu_images, cpu_texts = data