当前位置：网站首页>Delete the weight of the head part of the classification network pre training weight and modify the weight name

Delete the weight of the head part of the classification network pre training weight and modify the weight name

2022-07-24 05:49:00 【Hehai CV chicken with vegetables】

  Create a new dictionary , First the pretrained_dict Go through the key values in , Press . Division , If the first word in the weight name is not head, Put it into the new dictionary , At the same time, you can rename .

原网站

版权声明
本文为[Hehai CV chicken with vegetables]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/205/202207240517023991.html

边栏推荐

猜你喜欢

随机推荐