SavedModelBundle depolamadan yüklenen bir modeli temsil eder.
Model, hesaplamanın bir açıklamasından (bir Graph
), depoda kaydedilen değerlere başlatılan tensörlü bir Session
(örneğin, grafikteki parametreler veya değişkenler) ve modelin bir MetaGraphDef protokol arabelleği olarak bir açıklamasından oluşur.
İç İçe Sınıflar
sınıf | SavedModelBundle.Exporter | SavedModel'i dışa aktarma seçenekleri. | |
sınıf | SavedModelBundle.Loader | SavedModel yükleme seçenekleri. |
Sabitler
Sicim | DEFAULT_TAG |
Genel Yöntemler
Harita<Dize, Tensör > | |
geçersiz | |
statik SavedModelBundle.Exporter | ihracatçı (Dize ihracatDir) Kaydedilmiş bir modeli dışa aktarın. |
Beton Fonksiyonu | işlev (Dize imzaKey) İşlev imzasına karşılık gelen bir ConcreteFunction döndürün. |
Grafik | grafik () Model tarafından gerçekleştirilen hesaplamayı açıklayan grafiği döndürür. |
statik SavedModelBundle | load (String importDir, String... etiketleri) Kaydedilmiş bir modeli bir dışa aktarma dizininden yükleyin. |
statik SavedModelBundle.Loader | yükleyici (Dize ihracatDir) Kaydedilmiş bir modeli yükleyin. |
MetaGraphDef | metaGraphDef () Kaydedilen modelle ilişkili MetaGraphDef protokol arabelleğini döndürür. |
Oturum | |
Liste< İmza > | imzalar () Bu kayıtlı modelde bulunan tüm işlevlerin imzasını döndürün. |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize DEFAULT_TAG
Genel Yöntemler
genel Harita<String, Tensor > çağrısı (Map<String, Tensor > argümanlar)
Varsayılan işlevi doğrudan bu modelden çağırır.
Varsayılan işlev seçimi, aşağıdaki koşullardan ilkinin doğru olması temel alınarak yapılır:
- İşlev, bu kaydedilen modelin ana grafiğine eklenen tek imzadır
- İşlev, "serving_default" olan varsayılan imza adıyla eşlenir
Arayan, geri gönderilen tüm Tensörlerin kapatılmasından sorumludur.
Parametreler
argümanlar | imza adlarına göre eşlenen giriş tensörlerinin listesi |
---|
İade
- imza adına göre eşlenen çıktı tensörlerinin listesi
Atar
YasadışıTartışmaİstisna | varsayılan olarak hiçbir işlev seçilemiyorsa |
---|
genel boşluk kapat ()
genel statik SavedModelBundle.Exporter dışa aktarıcı (Dize dışa aktarmaDir)
Kaydedilmiş bir modeli dışa aktarın.
Modeli gerçekten kaydetmeden önce yapılandırma seçeneklerini ayarlamak için bir Exporter
nesnesi döndürür.
Parametreler
ihracatDizini | kayıtlı bir modeli içeren dizin yolu. |
---|
genel ConcreteFunction işlevi (Dize imzaKey)
İşlev imzasına karşılık gelen bir ConcreteFunction
döndürün.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
Parametreler
imzaAnahtar | Kaydedilen modeldeki SignatureDef adı. |
---|
İade
- bir işleve çağrı yapmak için kullanılabilecek nesne
Atar
YasadışıTartışmaİstisna | Bu kayıtlı signatureKey bulunamazsa. |
---|
genel statik SavedModelBundle yükü (String importDir, String... etiketleri)
Kaydedilmiş bir modeli bir dışa aktarma dizininden yükleyin. Yüklenmekte olan model, Kaydedilen Model API'si kullanılarak oluşturulmalıdır.
Bu yöntem aşağıdakilerin kısaltmasıdır:
SavedModelBundle.loader().withTags(tags).load();
Parametreler
ihracatDizini | kayıtlı bir modeli içeren dizin yolu. |
---|---|
Etiketler | yüklenecek belirli metagraf tanımını tanımlayan etiketler. |
İade
- grafiği ve ilgili oturumu içeren bir paket.
genel statik SavedModelBundle.Loader yükleyici (String ExportDir)
Kaydedilmiş bir modeli yükleyin.
Modeli gerçekten yüklemeden önce yapılandırma seçeneklerini ayarlayabilen bir Loader
nesnesi döndürür.
Parametreler
ihracatDizini | kayıtlı bir modeli içeren dizin yolu. |
---|
genel MetaGraphDef metaGraphDef ()
Kaydedilen modelle ilişkili MetaGraphDef protokol arabelleğini döndürür.
herkese açık Oturum oturumu ()
Model kullanılarak hesaplamanın gerçekleştirileceği Session
döndürür.
İade
- başlatılan oturum