no message

BlinkDL · Oct 27, 2021 · ccc9a05 · ccc9a05
1 parent e1babeb
commit ccc9a05
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/run.py b/run.py
@@ -72,7 +72,7 @@
 train_dataset.itos = {int(k): v for k, v in word_table.items()}
 UNKNOWN_CHAR = train_dataset.stoi['\ue083']
 
-print('\nLoading model...', end=' ')
+print(f'\nLoading model for {RUN_DEVICE}...', end=' ')
 if RUN_DEVICE == 'dml':
     import onnxruntime as rt
     sess_options = rt.SessionOptions()

diff --git a/server.py b/server.py
@@ -168,7 +168,7 @@ def NeuralWorker(queueZ, queueX):
     print('请关注我的知乎 https://zhuanlan.zhihu.com/p/394766831')
     print('\n声明：模型的训练数据全部来自网文，缺乏生活常识。生成的文字仅供娱乐。请遵守法律法规。')
 
-    print('loading model...')
+    print(f'\nLoading model for {RUN_DEVICE}...', end=' ')
 
     with open(WORD_NAME + '.json', "r", encoding="utf-16") as result_file:
         word_table = json.load(result_file)