SavedModelBundle

klasa publiczna SavedModelBundle

SavedModelBundle reprezentuje model załadowany z magazynu.

Model składa się z opisu obliczeń ( Graph ), Session z tensorami (np. parametrami lub zmiennymi na wykresie) zainicjowanymi do wartości zapisanych w pamięci oraz opisu modelu jako bufora protokołu MetaGraphDef .

Klasy zagnieżdżone

klasa SavedModelBundle.Exporter Opcje eksportowania SavedModel.
klasa SavedModelBundle.Loader Opcje ładowania SavedModel.

Stałe

Smyczkowy DOMYŚLNY_TAG

Metody publiczne

Mapa<String, Tensor >
wywołanie (Map<String, Tensor > argumenty)
Wywołuje funkcję domyślną bezpośrednio z tego modelu.
próżnia
zamknąć ()
Zwalnia zasoby ( Graph i Session ) powiązane z zapisanym pakietem modelu.
statyczny SavedModelBundle.Exporter
eksporter (String eksportDir)
Eksportuj zapisany model.
Funkcja betonu
funkcja (String podpisKey)
Zwróć ConcreteFunction odpowiadającą sygnaturze funkcji.
Wykres
wykres ()
Zwraca wykres opisujący obliczenia wykonane przez model.
statyczny plik SavedModelBundle
załaduj (String eksportDir, String... tagi)
Załaduj zapisany model z katalogu eksportu.
statyczny SavedModelBundle.Loader
moduł ładujący (String eksportDir)
Załaduj zapisany model.
MetaGraphDef
metaGraphDef ()
Zwraca bufor protokołu MetaGraphDef powiązany z zapisanym modelem.
Sesja
sesja ()
Zwraca Session , za pomocą której należy wykonać obliczenia przy użyciu modelu.
Lista <Podpis>
podpisy ()
Zwróć sygnaturę wszystkich funkcji dostępnych w tym zapisanym modelu.

Metody dziedziczone

Stałe

publiczny statyczny końcowy ciąg znaków DEFAULT_TAG

Wartość stała: „podaj”

Metody publiczne

public Map<String, Tensor > wywołanie (Map<String, Tensor > argumenty)

Wywołuje funkcję domyślną bezpośrednio z tego modelu.

Domyślny wybór funkcji odbywa się na podstawie pierwszego z poniższych warunków, który jest spełniony:

  • Funkcja jest jedyną dostępną sygnaturą dołączoną do głównego wykresu zapisanego modelu
  • Funkcja jest mapowana na domyślną nazwę podpisu, czyli „serving_default”

Osoba dzwoniąca jest odpowiedzialna za zamknięcie wszystkich zwróconych tensorów.

Parametry
argumenty lista tensorów wejściowych, odwzorowana według ich nazwy podpisu
Powroty
  • lista tensorów wyjściowych, odwzorowanych według nazwy podpisu
Rzuca
Wyjątek IllegalArgument jeśli domyślnie nie można wybrać żadnej funkcji

publiczna pustka zamknij ()

Zwalnia zasoby ( Graph i Session ) powiązane z zapisanym pakietem modelu.

public static SavedModelBundle.Exporter eksporter (String eksportDir)

Eksportuj zapisany model.

Zwraca obiekt Exporter do ustawiania opcji konfiguracyjnych przed faktycznym zapisaniem modelu.

Parametry
eksportDir ścieżka katalogu zawierającego zapisany model.

publiczna funkcja ConcreteFunction (String podpisKey)

Zwróć ConcreteFunction odpowiadającą podpisowi funkcji.

ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
 Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);

Parametry
podpisKlucz nazwa SignatureDef w zapisanym modelu.
Powroty
  • obiekt, którego można użyć do wywołania funkcji
Rzuca
Wyjątek IllegalArgument jeśli w zapisanym modelu nie zostanie znaleziony signatureKey .

publiczny wykres wykresu ()

Zwraca wykres opisujący obliczenia wykonane przez model.

publiczne statyczne ładowanie pakietu SavedModelBundle (tagi String eksportDir, String...)

Załaduj zapisany model z katalogu eksportu. Ładowany model należy utworzyć przy użyciu API Saved Model .

Ta metoda jest skrótem od:

SavedModelBundle.loader().withTags(tags).load();

Parametry
eksportDir ścieżka katalogu zawierającego zapisany model.
tagi znaczniki identyfikujące konkretny metagraphdef do załadowania.
Powroty
  • pakiet zawierający wykres i powiązaną sesję.

public static SavedModelBundle.Loader moduł ładujący (String eksportDir)

Załaduj zapisany model.

Zwraca obiekt Loader , który może ustawić opcje konfiguracyjne przed faktycznym załadowaniem modelu,

Parametry
eksportDir ścieżka katalogu zawierającego zapisany model.

publiczny MetaGraphDef metaGraphDef ()

Zwraca bufor protokołu MetaGraphDef powiązany z zapisanym modelem.

sesja publiczna sesja ()

Zwraca Session , za pomocą której należy wykonać obliczenia przy użyciu modelu.

Powroty
  • zainicjowanej sesji

publiczna Lista< Podpis > podpisy ()

Zwróć sygnaturę wszystkich funkcji dostępnych w tym zapisanym modelu.