SavedModelBundle مدلی را نشان می دهد که از فضای ذخیره سازی بارگیری شده است.
این مدل شامل توصیفی از محاسبات (یک Graph
)، یک Session
با تانسورها (مثلاً پارامترها یا متغیرهای موجود در نمودار) است که به مقادیر ذخیره شده در ذخیره سازی مقداردهی اولیه شده است، و توصیفی از مدل به عنوان یک بافر پروتکل MetaGraphDef .
کلاس های تو در تو
کلاس | SavedModelBundle.Exporter | گزینه هایی برای صادرات SavedModel. | |
کلاس | SavedModelBundle.Loader | گزینه هایی برای بارگیری SavedModel. |
ثابت ها
رشته | DEFAULT_TAG |
روش های عمومی
نقشه< رشته، تانسور > | |
باطل | |
استاتیک SavedModelBundle.Exporter | صادرکننده (String exportDir) یک مدل ذخیره شده را صادر کنید. |
ConcreteFunction | تابع (String signatureKey) یک ConcreteFunction مربوط به امضای تابع را برگردانید. |
نمودار | نمودار () نموداری را که محاسبات انجام شده توسط مدل را توصیف می کند، برمی گرداند. |
SavedModelBundle ثابت | بارگذاری (برچسب های رشته exportDir، رشته...) یک مدل ذخیره شده را از دایرکتوری صادرات بارگیری کنید. |
استاتیک SavedModelBundle.Loader | لودر (String exportDir) یک مدل ذخیره شده را بارگیری کنید. |
MetaGraphDef | متا گراف دف () بافر پروتکل MetaGraphDef مرتبط با مدل ذخیره شده را برمی گرداند. |
جلسه | |
فهرست< امضا > | امضاها () امضای همه عملکردهای موجود در این مدل ذخیره شده را برگردانید. |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی DEFAULT_TAG
روش های عمومی
نقشه عمومی <رشته، تنسور > فراخوانی (نقشه<رشته، تنسور > آرگومانها)
تابع پیش فرض را مستقیماً از این مدل فراخوانی می کند.
انتخاب تابع پیش فرض بر اساس اولین شرط زیر انجام می شود که درست است:
- تابع تنها امضای موجود است که به نمودار اصلی این مدل ذخیره شده پیوست شده است
- تابع به نام امضای پیشفرض نگاشت میشود که "serving_default" است.
تماس گیرنده مسئول بستن تمام تنسورهای برگشتی است.
پارامترها
استدلال ها | فهرستی از تانسورهای ورودی که با نام امضای آنها نگاشت شده است |
---|
برمی گرداند
- فهرستی از تانسورهای خروجی، که با نام امضا نگاشت شده است
پرتاب می کند
IllegalArgumentException | اگر هیچ عملکردی به طور پیش فرض انتخاب نشود |
---|
صادرکننده عمومی static SavedModelBundle.Exporter (String exportDir)
یک مدل ذخیره شده را صادر کنید.
قبل از ذخیره واقعی مدل، یک شی Exporter
را برای تنظیم گزینههای پیکربندی برمیگرداند.
پارامترها
exportDir | مسیر دایرکتوری حاوی یک مدل ذخیره شده |
---|
تابع ConcreteFunction عمومی (String signatureKey)
یک ConcreteFunction
مربوط به امضای تابع را برگردانید.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
پارامترها
کلید امضا | نام SignatureDef در مدل ذخیره شده. |
---|
برمی گرداند
- شیئی که می تواند برای برقراری تماس با یک تابع استفاده شود
پرتاب می کند
IllegalArgumentException | اگر signatureKey در این مدل ذخیره شده یافت نشد. |
---|
بارگذاری عمومی استاتیک SavedModelBundle (برچسبهای رشته exportDir، رشته...)
یک مدل ذخیره شده را از دایرکتوری صادرات بارگیری کنید. مدلی که در حال بارگیری است باید با استفاده از Saved Model API ایجاد شود.
این روش مختصری است برای:
SavedModelBundle.loader().withTags(tags).load();
پارامترها
exportDir | مسیر دایرکتوری حاوی یک مدل ذخیره شده |
---|---|
برچسب ها | برچسب هایی که متاگراف دف خاصی را برای بارگذاری شناسایی می کنند. |
برمی گرداند
- یک بسته حاوی نمودار و جلسه مربوطه.
بارگذار عمومی استاتیک SavedModelBundle.Loader (String exportDir)
یک مدل ذخیره شده را بارگیری کنید.
یک شی Loader
را برمی گرداند که می تواند گزینه های پیکربندی را قبل از بارگیری واقعی مدل تنظیم کند.
پارامترها
exportDir | مسیر دایرکتوری حاوی یک مدل ذخیره شده |
---|
جلسه عمومی ()
Session
که با آن محاسبات را با استفاده از مدل انجام می دهد، برمی گرداند.
برمی گرداند
- جلسه اولیه