Closed Arafat4341 closed 3 years ago
I am trying to train model on my dataset. However at epoch-3 training doesn't go further. It remains still. Besides the code doesn't also utilize gpu.
Can anyone tell how to make it utilize GPU and accelerate training? Thanks in advance!
I am trying to train model on my dataset. However at epoch-3 training doesn't go further. It remains still. Besides the code doesn't also utilize gpu.
Can anyone tell how to make it utilize GPU and accelerate training? Thanks in advance!