How to speed up the training?

[ZouhairBALLOUCH]

trafficstars

Hi ; I just want to know if there is a possibility to speed up the training because for me each epoch requires almost 4 hours of calculation Thanks in advance

[loki-veera]

Are you using GPU/CPU?. With Semantic KITTI on CPU it took around 4 hours per epoch but in GPU it was 30 min. check whether the code is using GPU or not.

[ARITRA2296]

I am unable to mount the process to GPU. Can you suggest to me the necessary changes so that I can mount it on the GPU??

[loki-veera]

@ARITRA2296 Did you install tensorflow or tensorflow-gpu. tensorflow-gpu should pick the GPU automatically. You can use something like this is https://stackoverflow.com/questions/38559755/how-to-get-current-available-gpus-in-tensorflow. To check your GPUs in the code.