Taking too much time in training
CS-savvy opened this issue · 3 comments
CS-savvy commented
Model taking 2 hrs for one epoch having 2300 images and batch size is 1, but you guys have mentioned it took only 4 hrs to train page detection model which contains 1600 images for 30 epochs.
can someone tell me the reason?
solivr commented
Can you give your GPU specs ?
CS-savvy commented
Thanks for replying @solivr
I am using Azure VM to train dhSegment model having NVIDIA-driver 390.116 GPU - Tesla K80 - 11 Gb gpu with 6 vcpu and 56 GB ram.