10 2020 档案

摘要:我自己改进的模型为model(model = ResNet(Bottleneck, [3, 4, 6, 3], **kwargs)),原模型为resnet50。 1.查看模型参数 现模型: 1 model_dict = model.state_dict() 2 for k,v in model_di 阅读全文
posted @ 2020-10-28 19:56 最咸的鱼 阅读(2455) 评论(0) 推荐(1) 编辑
摘要:详细理论部分可参考https://www.cnblogs.com/wanghui-garcia/p/10862733.html BCELoss()和BCEWithLogitsLoss()的输出logits和目标labels(必须是one_hot形式)的形状相同。 CrossEntropyLoss() 阅读全文
posted @ 2020-10-17 15:37 最咸的鱼 阅读(6243) 评论(0) 推荐(2) 编辑

点击右上角即可分享
微信分享提示