search

lmb-freiburg / demon

DeMoN: Depth and Motion Network
https://lmb.informatik.uni-freiburg.de/people/ummenhof/depthmotionnet/
GNU General Public License v3.0
574 stars 156 forks source link

Training using CPU #27

Open Narasappapraveen opened 6 years ago

Narasappapraveen commented 6 years ago

Hi, Is it possible to train the DEMON using CPU??

If yes, what changes I need to do in the code??

benjaminum commented 6 years ago

It is possible, but we did not test it. A problem is the tensor data format. Our custom ops that we use e.g. flow_to_depth2 and depth_to_flow only support the NCHW format, but CPU training is only possible with NHWC. We have added format conversion ops in the v2 training code like in the inference example but we never tested the training process.