如何把imagenet預(yù)訓(xùn)練的模型,輸入層的通道數(shù)隨心所欲的修改,從而來(lái)適應(yīng)自己的任務(wù) 1 2 3 4 5 6 7 8 9 10 11 12 13 14 #增加一個(gè)通道 w=layers[0].weight layers[0]=nn.Conv2d(4,64, kernel_size=(7,7), stride=(2,2), padding=(3,3), bias=False) layers[0].weight=torch.nn.Parameter...
www.dbjr.com.cn/article/1779...htm 2025-5-28