Tensorflow-SegNet icon indicating copy to clipboard operation
Tensorflow-SegNet copied to clipboard

Published 20 hours ago • verified publisher icon tkuanlun350

Run on the CPU

Open DenisN03 opened this issue 7 years ago • 4 comments

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

DenisN03 avatar Jan 11 '18 12:01 DenisN03

I would also like to know the answer to this. When I try to run it on the CPU, I always get a

No OpKernal was registered to support Op 'MaxPoolWithArgmax' with these attrs. Registered devices: [CPU], Registered kernels: <no registered kernels>

Or

KeyError: "Registering two gradient with name 'MaxPoolWithArgmax' !(Previous registration was in _find_and_load_unlocked <frozen importlib._bootstrap>:955)"

error

tom-bu avatar Jan 22 '18 07:01 tom-bu

hi, I also have this problem, does anybody know how to run it on th CPU

ansj11 avatar Feb 27 '18 06:02 ansj11

Well, me too. Can I run this model on cpu?

weiyao1996 avatar Mar 25 '18 07:03 weiyao1996

you guys can try to run it on the floydhub if you don't have a gpu, that's my solution

awesome-jeremy avatar May 06 '18 12:05 awesome-jeremy