参数量与计算量这个文件的显示错误修正 #845
Open
参数量与计算量这个文件的显示错误修正 #845
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
w5688414
reviewed
Feb 13, 2022
| $$Param_{linear} = C_{in} * C_{out} + C_{out} $$。我们这里打印一下线性层参数的形状看看。 | ||
|
|
||
| $Param_{linear} = C_{in} * C_{out} + C_{out}$。 | ||
|
|
| @@ -16,12 +16,16 @@ shape of weight: (4, 2, 3, 3) | |||
| shape of bias: (4,) | |||
| ``` | |||
| 这里解释一下上面的代码，我们先定义了一个卷积层cv2d，然后输出了这个卷积层的参数的形状，参数包含两部分，分别是weight和bias，这两部分相加才是整个卷积的参数量。因此，可以看到，我们定义的cv2d的参数量为：$4*2*3*3+4 = 76$, 4对应的是输出的通道数，2对应的是输入的通道数，两个3是卷积核的尺寸，最后的4就是bias的数量了， 值得注意的是， bias是数量与输出的通道数保持一致。因此，我们可以得出，一个卷积层的参数量的公式，如下： | |||
| $$ Param_{conv2d} = C_{in} * C_{out} * K_h * K_w + C_{out} $$ | |||
|
|
|||
There was a problem hiding this comment.
这样的公式打双$符号
然后公式的上下空出一行，公式就可以单独成行
change according the suggestions
127行去掉python
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
修改这个文件的部分显示错误