search

aws / sagemaker-mxnet-training-toolkit

Toolkit for running MXNet training scripts on SageMaker. Dockerfiles used for building SageMaker MXNet Containers are at https://github.com/aws/deep-learning-containers.
Apache License 2.0
60 stars 55 forks source link

Fix serving, update integ tests, and allow user-supplied transform_fn #40

Closed laurenyu closed 6 years ago

laurenyu commented 6 years ago

Description of changes: a few different things happened here:

  1. thanks to aws/sagemaker-containers#122, detecting a transform_fn in the user module doesn't need to throw an error
  2. ^that let me unskip one of the integ tests
  3. then I realized I could have unskipped one of the integ tests in my last PR
  4. also I found some bugs in my previous serving code, so that's fixed here

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.