一台机器如何训练tensorflow yolov3哪个暗网层由cpu训练而yolo层由gpu训练

时间:2018-08-20 03:10:41

标签: tensorflow gpu cpu

我的机器是1050 GPU(2 G)和8 G CPU,当我全部通过GPU训练Yolo-V3模型时,我的机器会产生oom。所以我希望切口模型是两个模块,其中一个由CPU训练,另一个由GPU训练,怎么做

0 个答案:

没有答案