![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
函数解析
文章平均质量分 65
小刺猬69
这个作者很懒,什么都没留下…
展开
-
断点代码设置
断点代码设置1. 找到你的model所对应的类名称MixVisionTransformer:如下class MixVisionTransformer(nn.Module): def __init__(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512], num_heads=[1, 2, 4, 8], mlp_ratio原创 2022-05-09 21:16:07 · 388 阅读 · 0 评论 -
nn.moduleList 和Sequential的理解
1.nn.ModuleList1.1 when如果在构造函数__init__中用到list、tuple、dict等对象时,一定要思考是否应该用ModuleList或ParameterList代替。如果你想设计一个神经网络的层数作为输入传递。1.2whatnn.ModuleList,它是一个储存不同 module,并自动将每个 module 的 parameters 添加到网络之中的容器。所有nn.ModuleList 内部的nn....原创 2022-04-18 09:39:39 · 2689 阅读 · 1 评论