Thanks for your contribution!
But when I use the the command "python train.py" to run the model after all preprocessing procedures are done, my program is stuck without any messages shown on the screen. After a long time like a century, there is some information which is supposed to be printed.
So I have no idea which part takes much of time? Is the operation 'load_coco_data' that reads the .pkl file?