microsoft / AzureML-BERT

End-to-End recipes for pre-training and fine-tuning BERT using Azure Machine Learning Service
https://azure.microsoft.com/en-us/blog/microsoft-makes-it-easier-to-build-popular-language-representation-model-bert-at-large-scale/
MIT License
394 stars 127 forks source link

docker folder doesn't exist for pretrain #44

Open usuyama opened 4 years ago

usuyama commented 4 years ago

docker folder for docker file and instruction to use Azure Container Registry

https://github.com/microsoft/AzureML-BERT/blob/master/pretrain/PyTorch/README.md

It seems the custom image is not needed from this commit d49e7cd58ae68ff209e374fdb730d2cd4ef29521 , but would be great to know which Dockerfile was used for creating the public image.

@aashnamsft

usuyama commented 4 years ago

From the name, this (https://github.com/Azure/AzureML-Containers/blob/master/base/gpu/openmpi3.1.2-cuda10.0-cudnn7-ubuntu16.04/Dockerfile ) looks like the base image; however, it seems there're additional steps for the AzureML-BERT image.

Still looking for the Dockerfile.