pytorch/serve

[Suggestion] Loading torch-model-archiver arguments from YAML

Open

#780 aberto em 10 de nov. de 2020

Ver no GitHub
 (6 comments) (1 reaction) (1 assignee)Java (790 forks)batch import
enhancementgood first issuetriaged_wait

Métricas do repositório

Stars
 (3.844 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

Is your feature request related to a problem? Please describe.

torch-model-archiver takes in a lot of params that makes it a hefty command line argument

Describe the solution

Allowing all the param values to be parsed from a YAML file when torch-model-archiver is used with a --yaml flag. It will also help automating creation of multiple .mar from different configurations.

Describe alternatives solution

I've tried calling torch-model-archiver from a shell script with variable params and that works file as well. But I just wanted to bounce this idea off the community to see if it is something people could find useful. I'd be happy to work on it.

Guia do colaborador