在PyTorch项目中反复手动编写Conv2d、BatchNorm2d、Dropout等网络层组合时，最令人头疼的莫过于参数顺序错误、in_channels遗漏传递，或者stride与padding不匹配引发的RuntimeError。CodeGeex提供了一种更直接的方式：只需将目标网络结构以自然语言注释的形式描述出来，它便能自动生成结构正确、可直接嵌入模型类的nn.Sequential或自定义Module代码。下面从三个典型应用场景出发，详细拆解如何充分利用这一功能。

CodeGeex能够基于自然语言注释生成结构准确的PyTorch模型代码：首先编写注释描述网络结构，右键触发生成，随后校验层参数是否与上下文兼容，必要时追加约束提示词进行重试。

使用CodeGeex快速构建标准卷积块

这一场景非常适合搭建ResNet基础模块或CNN特征提取器——它能省去手动计算通道数与尺寸变化的繁琐步骤。

操作非常直观：在PyCharm或VSCode中新建Python文件，将光标置于空行，输入注释描述结构，例如“构建一个标准卷积块：3×3卷积→BatchNorm2d→ReLU→2×2最大池化，输入通道64，输出通道128，步长1，padding1”。选中该行注释，右键点击【Generate Code】。生成后检查代码中是否包含nn.Conv2d(64, 128, 3, stride=1, padding=1)与nn.MaxPool2d(2)。若池化层未能出现，说明提示词中“最大池化”未被正确识别，只需在注释末尾追加“必须包含nn.MaxPool2d”并重新生成即可。

为自定义网络结构注入上下文后再生成

当项目中已经定义了数据集类、预处理逻辑或特定损失函数时，直接生成的模型很可能与现有流程不兼容。此时必须明确告知CodeGeex相关的上下文信息。

一种有效方法是在待生成模型类的上方插入CONTEXT注释块。例如：

# CONTEXT: 输入张量形状为 [B, 3, 224, 224]；已定义 loss_fn = nn.CrossEntropyLoss(label_smoothing=0.1)；训练时调用 model(x).logits

然后另起一行写下需求：“基于CONTEXT，生成一个ViT-like分类模型，包含PatchEmbed、TransformerEncoder层，输出logits形状为[B, 1000]”。触发生成后，重点核验forward方法是否返回一个包含"logits"键的字典。若返回的是原始张量而非字典，后续的loss_fn将会报错——这是最容易踩坑的地方。

修复生成代码中常见的维度错误

CodeGeex有时会忽略输入通道与上一层输出通道之间的衔接关系，尤其是在多分支结构中更容易出现此类问题。

举例来说：生成的代码中若含有self.conv1 = nn.Conv2d(3, 64, 7, stride=2, padding=3)，且输入是经Normalize后的RGB图像float32张量，通道数3是合理的。但后续如果接上self.conv2 = nn.Conv2d(64, 128, 3)，则必须确保前一层输出确实是64通道。若前层使用了AdaptiveAvgPool2d缩减了通道数，这里的64就会导致报错。

解决办法是在提示词中强制约束：例如写成“conv2的in_channels必须等于conv1的out_channels，不允许修改数值”，然后重新生成。操作很简单，将修正后的提示词粘贴进去重试即可。