gpt4 book ai didi

How to add custom Layers in ConvNext Small?(如何在ConvNext Small中添加自定义层?)

转载 作者:bug小助手 更新时间:2023-10-22 14:29:41 34 4
gpt4 key购买 nike



How to add custom layers after specific layer of ConNext small?
below is the link of github code of ConvNext Small https://github.com/facebookresearch/ConvNeXt

如何在ConNext的特定图层后添加自定义图层?以下是ConvNext Small的github代码链接https://github.com/facebookresearch/ConvNeXt


更多回答

Can you be more specific regarding which layers you would like to add and where?

你能更具体地说明你想添加哪些层以及在哪里吗?

I added a block of layers at the end of 4th stage in ConvNext Small. Then i checked its layered architecture its added at the right place where i wanted but when I train it with pretrained weights the error comes RuntimeError: Error(s) in loading state_dict for ConvNeXt: Missing key(s) in state_dict: "custom_block.conv.weight", "custom_block.conv.bias", "custom_block.bn.weight", "custom_block.bn.bias", "custom_block.bn.running_mean", "custom_block.bn.running_var".

我在ConvNext Small的第四阶段结束时添加了一块图层。然后,我检查了它的分层架构,它添加在了我想要的正确位置,但当我用预先训练的权重训练它时,错误出现了RuntimeError:为ConvNeXt加载state_dict时出错:state_dict:“custom_block.conv.weight”、“custom_block.conv.bias”、“customer_block.bn.weight”、”custom_blok.bn.bias“、”custot_block.bn.running_mear“、”customer_block.bn.running_var“。

Can you share the code that you've used to add layers and load pretrained weights?

你能分享你用来添加层和加载预训练权重的代码吗?

优秀答案推荐
更多回答

34 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com