help wanted
描述
InternalError (see above for traceback): CUB reduce errorinvalid device function [[Node: gradients/vgg_16_2/dropout7/dropout/div_grad/Sum_1 = Sum[T=DT_FLOAT, Tidx=DT_INT32, keep_dims=false, _device="/job:localhost/replica:0/task:0/device:GPU:0"](gradients/vgg_16_2/dropout7/dropout/div_grad/mul, ConstantFolding/gradients/LOSS_default/mul_6_grad/BroadcastGradientArgs-1)]]
Command exited with non-zero status 1 17.37user 7.49system 0:17.64elapsed 140%CPU (0avgtext+0avgdata 3487244maxresident)k