Skip to content

Add a PyTorch worker #211

@ZchiPitt

Description

@ZchiPitt

Is your feature request related to a problem? Please describe.
We need to add a PyTorch worker that accepts the name of an PyTorch model file as an argument and compiles and evaluates it.

Describe the solution you'd like

  • install pytorch and make the build succeed
  • Get a small model () to run instead of the modified resnet50 of PtZendnn
  • add a test for PyTorch worker
  • add a new stage in the Dockerfile to build the PyTorch

Describe alternatives you've considered
NA

Additional context
NA

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions