weixin_45912366的博客

断点代码设置 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

1.nn.ModuleList 1.1 when 如果在构造函数__init__中用到list、tuple、dict等对象时，一定要思考是否应该用ModuleList或ParameterList代替。 如果你想设计一个神经网络的层数作为输入传递。 1.2what nn.ModuleList，它是一个储存不同 module，并自动将每个 module 的 parameters 添加到网络之中的容器。所有nn.ModuleList 内部的nn....