亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

“DataParallel”對象沒有屬性“conv1”

“DataParallel”對象沒有屬性“conv1”

慕尼黑8549860 2024-01-27 15:54:10
我正在嘗試conv1根據下面的代碼和架構可視化層的 cnn 網絡特征圖。它在沒有 DataParallel 的情況下正常工作,但是當我激活model = nn.DataParallel(model)它時,它會引發錯誤:“DataParallel”對象沒有屬性“conv1”。任何建議表示贊賞。class Model(nn.Module):    def __init__(self, kernel, num_filters, res = ResidualBlock):        super(Model, self).__init__()                self.conv0 = nn.Sequential(            nn.Conv2d(4, num_filters, kernel_size = kernel*3,                        padding = 4),            nn.BatchNorm2d(num_filters),            nn.ReLU(inplace=True))                self.conv1 = nn.Sequential(            nn.Conv2d(num_filters, num_filters*2, kernel_size = kernel,                       stride=2, padding = 1),            nn.BatchNorm2d(num_filters*2),            nn.ReLU(inplace=True))                self.conv2 = nn.Sequential(            nn.Conv2d(num_filters*2, num_filters*4, kernel_size = kernel, stride=2, padding = 1),            nn.BatchNorm2d(num_filters*4),            nn.ReLU(inplace=True))                       self.tsconv0 = nn.Sequential(            nn.ConvTranspose2d(num_filters*4, num_filters*2, kernel_size = kernel, padding = 1),            nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True),            nn.ReLU(inplace=True),            nn.BatchNorm2d(num_filters*2))        self.tsconv1 = nn.Sequential(            nn.ConvTranspose2d(num_filters*2, num_filters, kernel_size = kernel, padding = 1),            nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True),            nn.ReLU(inplace=True),            nn.BatchNorm2d(num_filters))                self.tsconv2 = nn.Sequential(            nn.Conv2d(num_filters, 1, kernel_size = kernel*3, padding = 4, bias=False),            nn.ReLU(inplace=True))model = Model(kernel, num_filters)model = nn.DataParallel(model)
查看完整描述

1 回答

?
料青山看我應如是

TA貢獻1772條經驗 獲得超8個贊

當您使用 時DataParallel,請在此處添加額外的內容module。而不是model.conv3.簡單地寫model.module.conv3.



查看完整回答
反對 回復 2024-01-27
  • 1 回答
  • 0 關注
  • 143 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號