It takes a long time to train semantic segmentation models

[moniec1717]

when I train my own dataset (point clouds of bridges and roads), it takes 15h for just one epoch.
How can I improve my model? Which parameters should I change?
I have already tried changing the parameter stride and block_size, but it didn't work on my dataset.
Can anyone help me out, thanks a lot!

[kvta1]

Are you training on GPU? You can increase batch_size

[moniec1717]

Are you training on GPU? You can increase batch_size

I've adjusted the batch size, but no great improvement. I thought maybe the GPU wasn't being used, but after checking the code and seeing that my computer’s memory did use it, I'm not really sure where else things could be improved.