segmentation
segmentation copied to clipboard
Run on the CPU
Can I run this model on cpu? What changes in the code is necessary to do this?
@fregu856
@DenisN03 just tried to do it on CPU after trying on my small GPU and having a memory problem.
The message I get is :
No OpKernel was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels:
@fregu856