eval with alphanumeric and case sensitive setting fix

frosenberg · Feb 25, 2020 · dff8448 · dff8448
1 parent 1a63bce
commit dff8448
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test.py b/test.py
@@ -141,6 +141,7 @@ def validation(model, criterion, evaluation_loader, converter, opt):
                 alphanumeric_case_insensitve = '0123456789abcdefghijklmnopqrstuvwxyz'
                 out_of_alphanumeric_case_insensitve = f'[^{alphanumeric_case_insensitve}]'
                 pred = re.sub(out_of_alphanumeric_case_insensitve, '', pred)
+                gt = re.sub(out_of_alphanumeric_case_insensitve, '', gt)
 
             if pred == gt:
                 n_correct += 1