Run on the CPU

[DenisN03]

Can I run this model on cpu? What changes in the code is necessary to do this?

[mrocr]

@fregu856

[UcefMountacer]

@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