use ignite for exp2

exp2/modified/model.py

@@ -21,9 +21,9 @@ class Cnn(torch.nn.Module):
         self.fc2 = torch.nn.Linear(64, 10)
 
         # 初始化模型参数
-        self._initialize_weights()
+        self.__initialize_weights()
 
-    def _initialize_weights(self):
+    def __initialize_weights(self):
         # YYC MARK:
         # 把两个全连接线性层按tensorflow默认设置初始化，即：
         # - kernel_initializer='glorot_uniform'