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
阅读全文
摘要:详细理论部分可参考https://www.cnblogs.com/wanghui-garcia/p/10862733.html BCELoss()和BCEWithLogitsLoss()的输出logits和目标labels(必须是one_hot形式)的形状相同。 CrossEntropyLoss()
阅读全文