标签: tensorflow keras yolo
我正在尝试使用keras和tensorflow在我的机器上测试yolo tiny v2,但是我有内存警告。 警告说:“分配器试图找到2.29GiB时内存不足。调用方表明这不是故障,但可能意味着如果有更多可用内存,则可能会提高性能。”但是,看起来有2.29GB的GPU内存对于微型yolo型号来说太大了。我正在将网络应用到摄像头的视频流中。我有2GB GPU Quadro K1000M。