Does the code runs on CPU?

[qsisi]

Hello! Thanks for open-sourcing this work.

When I run:
python train_synthetic_data.py --load_folder scripts/synthetic_data/line_pole_noise/

The observation is that the GPU is not actually used (only 300~MB occupied), but the CPU-util increases rapidly, so does the code runs on CPU instead of GPU?

I'm not familiar with tensorflow, could you please give some hints about it?

Thanks for your help.