aboutsummaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
Diffstat (limited to 'train.py')
-rwxr-xr-xtrain.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/train.py b/train.py
index 86e7337..25993f7 100755
--- a/train.py
+++ b/train.py
@@ -210,12 +210,12 @@ def main():
accuracy[M] = test_model(M, faces_train, faces_test, target_train, target_test, args)
end = timer()
print("Run with", M, "eigenvalues completed in ", end-start, "seconds")
-
+
else:
M = args.eigen
start = timer()
test_model(M, faces_train, faces_test, target_train, target_test, args)
end = timer()
-
+
if __name__ == "__main__":
main()