System Information
- Framework (e.g. TensorFlow) / Algorithm (e.g. KMeans): MXNet
- Framework Version: Latest
- Python Version: 3
- CPU or GPU: CPU and GPU
- Python SDK Version: Latest
- Are you using a custom image: No
Feature Request
It will be great to either specify the env dictionary or path to requirements file for the Sagemaker MxNet API.
|
def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_version='py2', |
It is already available in the Tensorflow API #52
We need to add additional dependencies and having the ability to specify requirements is much much easier than creating our own Docker image.
System Information
Feature Request
It will be great to either specify the env dictionary or path to requirements file for the Sagemaker MxNet API.
sagemaker-python-sdk/src/sagemaker/mxnet/estimator.py
Line 26 in 8b3d113
It is already available in the Tensorflow API #52
We need to add additional dependencies and having the ability to specify requirements is much much easier than creating our own Docker image.