do you mind to show your dockerfile building the image for training? I had hard time to get the environment setup. thank you