Pytorch
关注
分享
扫一扫
Pytorch的使用
微电子学与固体电子学-俞驰
如切如磋，如琢如磨，臻于至善。
展开
-
AttributeError: 'NoneType' object has no attribute '__array_interface__'
这个问题出现在dataloader里面.问题解决:原因是数据没有读进来.检查下dataloader的类中的读取数据的代码,例如:img = cv2.imread(img_name)读取失败的话,会得到None....原创 2019-11-01 15:30:11 · 7195 阅读 · 0 评论 -
torchvision中Transform的normalize
RGB還是GBR?根據[1][2],應該是:RGBtransforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])第一個中括號是RGB的mean第二個中括號是RGB的std上述代碼的具體用法在[3]中有完整代碼.Reference:[1]https://discuss.pyt...原创 2019-10-18 16:11:58 · 2227 阅读 · 1 评论 -
param.requires_grad = False的作用
代碼來自[1],[1]的詳細註釋在[2]中param.requires_grad = False的作用是:屏蔽預訓練模型的權重只訓練最後一層的全連接層的權重.最後一層的添加，是通過[2]中的一句代碼:model_conv.fc = nn.Linear(num_ftrs, 2)可以發現，凍結預訓練模型的權重後，驗證精度會下降．Reference:...原创 2019-10-18 15:17:20 · 6409 阅读 · 1 评论 -
BCEWithLogitsLoss的使用案例
在RSNA的比赛中的用法如下(GPU版本):torch.nn.BCEWithLogitsLoss(weight=torch.Tensor([2/7,1/7,1/7,1/7,1/7,1/7]).cuda())原创 2019-10-16 21:01:14 · 1392 阅读 · 0 评论