Closed athewsey closed 3 years ago
Tested successfully on these scenarios: Tensorflow Local Studio CPU kernel OK Tensorflow Sagemaker Studio CPU kernel OK PyTorch Local Studio CPU kernel OK PyTorch SageMaker Studio CPU kernel OK Tensorflow Local Notebook CPU instance OK Tensorflow Sagemaker Notebook CPU instance OK PyTorch Local Notebook CPU instance OK PyTorch SageMaker Notebook CPU instance OK
The prediction results are also reasonable.
The test does not include running it in local GPU environment for neither Studio nor Notebook.
The changes look good.
Issue #, if available: #1
Description of changes:
tokenize_*and*_pad_docs
function name and use same numpy one-hot implementation as PyTorch - rather than relying on TF dependency.By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.