修改教程文档

fangwei123456 · Sep 16, 2020 · 124b71f · 124b71f
1 parent c7b0d8a
commit 124b71f
Show file tree

Hide file tree

Showing 2 changed files with 179 additions and 180 deletions.
diff --git a/docs/source/clock_driven/4_conv_fashion_mnist.rst b/docs/source/clock_driven/4_conv_fashion_mnist.rst
@@ -59,10 +59,10 @@ ANN中常见的卷积神经网络，大多数是卷积+全连接层的形式，
 
     self.fc = nn.Sequential(
         nn.Flatten(),
-        layer.Dropout(0.7, behind_spiking_layer=True),
+        layer.Dropout(0.7, True),
         nn.Linear(128 * 7 * 7, 128 * 3 * 3, bias=False),
         neuron.LIFNode(tau=tau, v_threshold=v_threshold, v_reset=v_reset, surrogate_function=surrogate.ATan()),
-        layer.Dropout(0.7, behind_spiking_layer=True),
+        layer.Dropout(0.7, True),
         nn.Linear(128 * 3 * 3, 128, bias=False),
         neuron.LIFNode(tau=tau, v_threshold=v_threshold, v_reset=v_reset, surrogate_function=surrogate.ATan()),
         nn.Linear(128, 10, bias=False),
@@ -153,10 +153,10 @@ ANN中常见的卷积神经网络，大多数是卷积+全连接层的形式，
             )
             self.fc = nn.Sequential(
                 nn.Flatten(),
-                layer.Dropout(0.7, behind_spiking_layer=True),
+                layer.Dropout(0.7, True),
                 nn.Linear(128 * 7 * 7, 128 * 3 * 3, bias=False),
                 neuron.LIFNode(tau=tau, v_threshold=v_threshold, v_reset=v_reset, surrogate_function=surrogate.ATan()),
-                layer.Dropout(0.7, behind_spiking_layer=True),
+                layer.Dropout(0.7, True),
                 nn.Linear(128 * 3 * 3, 128, bias=False),
                 neuron.LIFNode(tau=tau, v_threshold=v_threshold, v_reset=v_reset, surrogate_function=surrogate.ATan()),
                 nn.Linear(128, 10, bias=False),