pytorch/serve

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

Open

#780 geöffnet am 10. Nov. 2020

Auf GitHub ansehen
 (6 Kommentare) (1 Reaktion) (1 zugewiesene Person)Java (790 Forks)batch import
enhancementgood first issuetriaged_wait

Repository-Metriken

Stars
 (3.844 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

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.

Contributor Guide