feat: 增加中文适配

521xueweihan · Feb 7, 2024 · 66fdb7f · 66fdb7f
1 parent ef309b1
commit 66fdb7f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/code/chapter-8/01_classification/utils/my_utils.py b/code/chapter-8/01_classification/utils/my_utils.py
@@ -18,6 +18,8 @@
 import torch.nn.init as init
 from datetime import datetime
 import logging
+import matplotlib
+matplotlib.rcParams['font.family'] = 'SimHei'
 
 
 class LeNet5(nn.Module):