search

aws-samples / amazon-sagemaker-bert-pytorch

MIT No Attribution
64 stars 25 forks source link

Use elastic inference for sagemaker v2 #9

Closed hongshanli23 closed 3 years ago

hongshanli23 commented 3 years ago

Issue #, if available:

Description of changes:

  1. Enforce torch 1.3.1 for model training
  2. Added batch inference
  3. Added input data sanity check

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.