-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Support for pytorch cli #1068
Copy link
Copy link
Closed
Labels
Description
Please fill out the form below.
System Information
- Framework: Pytorch
- Framework Version: 1.0, 1.1, 1.2
- Python Version: 3.6, 3.7
- CPU or GPU: both
- Python SDK Version: latest
- Are you using a custom image: No
Describe the problem
Current sagemaker cli doesn't support Pytorch (only MxNet and Tensorflow are supported), any plan to support this ? I don't mind to contribute though.
Reactions are currently unavailable