Training only uses CPU - Githubissues

aws-samples / amazon-sagemaker-tensorflow-object-detection-api

Train and deploy models using TensorFlow 2 with the Object Detection API on Amazon SageMaker

MIT No Attribution

44 stars 34 forks source link

Training only uses CPU #7

Closed t-T-s closed 3 years ago

t-T-s commented 3 years ago

The use_tpu is set to be False in by default. But even if we set it to be True, the tpu_resolver can not resolve the tpu from AWS sagemaker. How can I resolve the TPU for sagemaker ? In tf documentation we can only use that particular function in Google cloud only. Any workaround ?