[
https://issues.apache.org/jira/browse/BEAM-13984?focusedWorklogId=760004&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-760004
]
ASF GitHub Bot logged work on BEAM-13984:
-----------------------------------------
Author: ASF GitHub Bot
Created on: 21/Apr/22 13:00
Start Date: 21/Apr/22 13:00
Worklog Time Spent: 10m
Work Description: yeandy commented on code in PR #17196:
URL: https://github.com/apache/beam/pull/17196#discussion_r855159425
##########
sdks/python/setup.py:
##########
@@ -234,6 +234,10 @@ def get_version():
'azure-core >=1.7.0',
]
+ML_REQUIREMENTS = [
+ 'torch >= 1.10.2'
Review Comment:
For now, to allow most flexibility, we're going to support `pip install
apache-beam pytorch`
Issue Time Tracking
-------------------
Worklog Id: (was: 760004)
Time Spent: 3h 50m (was: 3h 40m)
> Implement RunInference for PyTorch
> ----------------------------------
>
> Key: BEAM-13984
> URL: https://issues.apache.org/jira/browse/BEAM-13984
> Project: Beam
> Issue Type: Sub-task
> Components: sdk-py-core
> Reporter: Andy Ye
> Assignee: Andy Ye
> Priority: P2
> Labels: run-inference
> Time Spent: 3h 50m
> Remaining Estimate: 0h
>
> Implement RunInference for PyTorch as described in the design doc
> [https://s.apache.org/inference-sklearn-pytorch]
> There will be a pytorch_impl.py file that contains PyTorchModelLoader and
> PyTorchInferenceRunner classes.
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
