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