Run on the CPU - Githubissues

fregu856 / segmentation

TensorFlow implementation of ENet, trained on the Cityscapes dataset.

http://www.fregu856.com/

MIT License

244 stars 73 forks source link

Run on the CPU #4

Open DenisN03 opened 6 years ago

DenisN03 commented 6 years ago

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

mrocr commented 5 years ago

@fregu856

UcefMountacer commented 4 years ago

@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