RuntimeError: CUDA error: CUBLAS

RuntimeError: CUDA error: CUBLAS,第1张

RuntimeError: CUDA error: CUBLAS

backbone = resnet50_fpn_backbone(norm_layer=torch.nn.BatchNorm2d,
trainable_layers=3)
将trainable_layers 改成4即可

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5595487.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-15
下一篇 2022-12-15

发表评论

登录后才能评论

评论列表(0条)

保存