Merge pull request #54 from HenryMakerCH/4.0

Update train.py 更改numba的logging优先级
2023-03-19 00:58:32 +08:00 · 2023-03-19 00:58:32 +08:00 · 447d95bc82
parent 293fe0bc03 758109d47e
commit 447d95bc82
1 changed files with 2 additions and 0 deletions
--- 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