diff --git a/train.py b/train.py
index e499528..c4ba1a1 100644
--- a/train.py
+++ b/train.py
@@ -3,6 +3,8 @@ import multiprocessing
 import time
 
 logging.getLogger('matplotlib').setLevel(logging.WARNING)
+logging.getLogger('numba').setLevel(logging.WARNING)
+
 import os
 import json
 import argparse