├── .DS_Store ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── HFbertdemo.ipynb ├── LICENSE ├── README.md ├── inference_scripts ├── .DS_Store ├── inference_inf1.py └── requirements.txt ├── traceutils.py └── train_scripts ├── .DS_Store ├── bert_train_torchrun_trn1.py └── requirements.txt /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/.DS_Store -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /HFbertdemo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/HFbertdemo.ipynb -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/README.md -------------------------------------------------------------------------------- /inference_scripts/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/inference_scripts/.DS_Store -------------------------------------------------------------------------------- /inference_scripts/inference_inf1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/inference_scripts/inference_inf1.py -------------------------------------------------------------------------------- /inference_scripts/requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==4.6.0 -------------------------------------------------------------------------------- /traceutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/traceutils.py -------------------------------------------------------------------------------- /train_scripts/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/train_scripts/.DS_Store -------------------------------------------------------------------------------- /train_scripts/bert_train_torchrun_trn1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aws-samples/sagemaker-trainium-inferentia/HEAD/train_scripts/bert_train_torchrun_trn1.py -------------------------------------------------------------------------------- /train_scripts/requirements.txt: -------------------------------------------------------------------------------- 1 | torchvision==0.12.0 2 | datasets==2.5.2 3 | transformers==4.17.0 4 | --------------------------------------------------------------------------------