Skip to content

Torch support in ML.NET #4053

Closed
Closed

Description

@artidoro

As ML.NET increases it's support for deep learning, we should add support for PyTorch models.
TorchSharp has bindings for LibTorch (PyTorch's C++ frontend) and offers the following functionality:

  • Score models that were previously defined and trained in PyTorch's python API
  • Define and train models directly in C#

We could have a scoring functionality similar to the TensorFlow scoring. We could also add training of Torch models defined in C#.

Activity

  1. GeorgeS2019 commented on Jul 12, 2022

    @GeorgeS2019

    Now we have the official Microsoft.ML.TorchSharp namespace.

    We could close this issue now :-)

  2. michaelgsharp commented on Jul 13, 2022

    @michaelgsharp
    Contributor

    @GeorgeS2019 While we do have the namespace, it currently only has a single TextClassification API. We want to add in support to do more of what is mentioned in this issue like defining TorchSharp models in C# and doing the full training. We are also tracking this elsewhere so we can close this issue, but we don't have all the features yet this issue mentions.

    @artidoro good to see you are still around and paying attention to github notifications :)

  3. artidoro commented on Jul 13, 2022

    @artidoro
    ContributorAuthor

    @michaelgsharp it's good to see Torch being integrated in ML.NET!
    Hope everything is going well :)

  4. GeorgeS2019 commented on Jul 14, 2022

    @GeorgeS2019

    @michaelgsharp

    it currently only has a single TextClassification API.

    Based on the Single example, it seems to me that it is doable to port TorchAudio and TorchVision examples from TorchSharp to ML.NET using the same way how the TexTclassification API is set up. This is just a thought, not sure if this is that straight forwards?

  5. ghost locked as resolved and limited conversation to collaborators on Aug 14, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions