pytorch/serve

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

Open

#780 ouverte le 10 nov. 2020

Voir sur GitHub
 (6 commentaires) (1 réaction) (1 assigné)Java (790 forks)batch import
enhancementgood first issuetriaged_wait

Métriques du dépôt

Stars
 (3 844 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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.

Guide contributeur