It seems work now but the training process is taking a lot of time, it’s very slow, when I searched I found that BERT method is in general slow, but in this case it may take hours or may be days, any idea ?