iTPN
iTPN copied to clipboard
fast_itpn
请问,有开源fast_itpn预训练代码的计划吗?
可以开源。等我有空整理一下
@sunsmarterjie 期待您的开源。还有个问题,fast_itpn目录下开源权重,是预训练权重,还是imagenet微调后的权重?
预训练权重
---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年01月15日 16:13 | | 收件人 | @.> | | 抄送至 | @.>@.> | | 主题 | Re: [sunsmarterjie/iTPN] fast_itpn (Issue #12) |
@sunsmarterjie 期待您的开源。还有个问题,fast_itpn目录下开源权重,是预训练权重,还是imagenet微调后的权重?
— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>
@sunsmarterjie 在fast_itpn里面,预训练模型转为检测模型backbone,需要做哪些修改呢(ps:检测的微调代码可以开源吗)
替换你想用的检测器的backbone就可以了。检测代码正在整理
---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年01月16日 11:38 | | 收件人 | @.> | | 抄送至 | @.>@.> | | 主题 | Re: [sunsmarterjie/iTPN] fast_itpn (Issue #12) |
@sunsmarterjie 在fast_itpn里面,预训练模型转为检测模型backbone,需要做哪些修改呢(ps:检测的微调代码可以开源吗)
— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>
@sunsmarterjie 我在下游任务使用 fast_itpn
目录下 modeling_finetune.py
中 Fast_iTPN
作为backbone
加载预训练权重时,发现权重里有 mask_token', 'align_dim_16tofpn.weight', 'align_dim_16tofpn.bias', 'fpn_modules.0.norm2.weight', ...
,
但是 Fast_iTPN
中没有定义,请问需要怎么修改呢?
直接忽略就行。这些是预训练用到的参数,下游微调不需要加载
好的,感谢感谢。