recv_device="/job:localhost/replica:0/task:0/device:CPU:0"

出现此类问题,我是采用过两种解决办法

第一种,我这里之前设置的只占百分之五十后来增加到八十,但作用大不大我也不清楚,主要靠第二种方法

recv_device=

第二种比较常用

读入batch_size减小点,比如之前设置的是64,后来改为32或者16

你可能感兴趣的:(recv_device="/job:localhost/replica:0/task:0/device:CPU:0")