记录在mmsegmentation框架里写代码跑实验时出现的bug

本文记录了在使用mmsegmentation框架进行深度学习实验时遇到的错误,包括尺寸匹配问题、键错误、数据加载错误、梯度问题、属性错误等,并给出了相应的解决策略。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1. ValueError: size shape must match input shape. Input is 2D, size is 1

这种是尺寸没对上,用input.shape和targe.shape把尺寸弄清楚,我缺的是通道那一维,用torch.unsqueeze(input, 1)就行了

2. KeyError: ‘layers.0.blocks.0.attn.relative_position_bias_table’

这通常是别人的权重拿到自己的网络没转换的原因

3. KeyError: ‘data_time’

mmseg在保存权重时出现的问题
mmseg/apis/train.py里,加上priority='LOW'
同时,如果想保存最佳结果,可以加上save_best='mIoU',或其他指标

runner.register_hook(eval_hook(val_dataloader
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值