損失函數Target size must be the same as input size

出錯程序 criterion_modality = torch.nn.BCEWithLogitsLoss() label = Variable(label.cuda()) loss = criterion_modality(outRGB, label) # 出錯行 問題 File "C:\Users\Rain\AppData\Local\Programs\Python\Anaconda.3.5
相關文章
相關標籤/搜索