Training using CPU

[Narasappapraveen]

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

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

[benjaminum]

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.