1 files changed, 1 insertions, 0 deletions

diff --git a/train.py b/train.py
index 3a15fa0..c5a72c2 100755
--- a/train.py
+++ b/train.py
@@ -288,3 +288,4 @@ def main():
     draw_results(args, target_test, target_pred)
 
 if __name__ == "__main__":
+    main()