backbone = resnet50_fpn_backbone(norm_layer=torch.nn.BatchNorm2d,
trainable_layers=3)
将trainable_layers 改成4即可
欢迎分享,转载请注明来源:内存溢出
backbone = resnet50_fpn_backbone(norm_layer=torch.nn.BatchNorm2d,
trainable_layers=3)
将trainable_layers 改成4即可
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)