SavedModelBundle แสดงถึงโมเดลที่โหลดจากที่จัดเก็บข้อมูล
แบบจำลองประกอบด้วยคำอธิบายของการคำนวณ ( Graph
) Session
ที่มีเทนเซอร์ (เช่น พารามิเตอร์หรือตัวแปรในกราฟ) กำหนดค่าเริ่มต้นเป็นค่าที่บันทึกไว้ในที่จัดเก็บ และคำอธิบายของแบบจำลองเป็น บัฟเฟอร์โปรโตคอล MetaGraphDef
คลาสที่ซ้อนกัน
ระดับ | SavedModelBundle ผู้ส่งออก | ตัวเลือกสำหรับการส่งออก SavedModel | |
ระดับ | SavedModelBundle.Loader | ตัวเลือกสำหรับการโหลด SavedModel |
ค่าคงที่
สตริง | DEFAULT_TAG |
วิธีการสาธารณะ
แผนที่<สตริง เทนเซอร์ > | |
เป็นโมฆะ | |
SavedModelBundle.Exporter แบบคงที่ | ผู้ส่งออก (สตริงexportDir) ส่งออกโมเดลที่บันทึกไว้ |
คอนกรีตฟังก์ชั่น | ฟังก์ชั่ น (String SignatureKey) ส่งกลับ ConcreteFunction ที่สอดคล้องกับลายเซ็นของฟังก์ชัน |
กราฟ | กราฟ () ส่งกลับกราฟที่อธิบายการคำนวณที่ทำโดยแบบจำลอง |
SavedModelBundle แบบคงที่ | โหลด (สตริง ExportDir, สตริง... แท็ก) โหลดโมเดลที่บันทึกไว้จากไดเร็กทอรีการส่งออก |
SavedModelBundle.Loader แบบคงที่ | ตัวโหลด (สตริงส่งออกDir) โหลดโมเดลที่บันทึกไว้ |
MetaGraphDef | เมตากราฟDef () ส่งกลับ บัฟเฟอร์โปรโตคอล MetaGraphDef ที่เกี่ยวข้องกับแบบจำลองที่บันทึกไว้ |
การประชุม | |
รายการ< ลายเซ็น > | ลายเซ็น () ส่งกลับลายเซ็นของฟังก์ชันทั้งหมดที่มีอยู่ในโมเดลที่บันทึกไว้นี้ |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ DEFAULT_TAG
วิธีการสาธารณะ
แผนที่สาธารณะ <สตริง, เทนเซอร์ > โทร (อาร์กิวเมนต์แผนที่ <สตริง, เทนเซอร์ >)
เรียกใช้ฟังก์ชันเริ่มต้นโดยตรงจากโมเดลนี้
การเลือกฟังก์ชันเริ่มต้นจะเสร็จสิ้นตามเงื่อนไขแรกที่เป็นจริงต่อไปนี้:
- ฟังก์ชันนี้เป็นลายเซ็นเดียวที่แนบกับกราฟหลักของแบบจำลองที่บันทึกไว้นี้
- ฟังก์ชันนี้ถูกแมปกับชื่อลายเซ็นเริ่มต้น ซึ่งก็คือ "serving_default"
ผู้เรียกมีหน้าที่ปิดเทนเซอร์ที่ส่งคืนทั้งหมด
พารามิเตอร์
ข้อโต้แย้ง | รายการเทนเซอร์อินพุต แมปตามชื่อลายเซ็น |
---|
การส่งคืน
- รายการเทนเซอร์เอาท์พุต แมปตามชื่อลายเซ็น
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | หากไม่มีฟังก์ชันใดสามารถเลือกได้ตามค่าเริ่มต้น |
---|
ผู้ส่งออก SavedModelBundle.Exporter สาธารณะแบบคงที่ (String ExportDir)
ส่งออกโมเดลที่บันทึกไว้
ส่งคืนออบเจ็กต์ Exporter
สำหรับการตั้งค่าตัวเลือกการกำหนดค่าก่อนที่จะบันทึกโมเดลจริง
พารามิเตอร์
ส่งออกDir | เส้นทางไดเร็กทอรีที่มีโมเดลที่บันทึกไว้ |
---|
ฟังก์ชัน ConcreteFunction สาธารณะ (String SignatureKey)
ส่งกลับ ConcreteFunction
ที่สอดคล้องกับลายเซ็นของฟังก์ชัน
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
พารามิเตอร์
ลายเซ็นคีย์ | ชื่อของ SignatureDef ในโมเดลที่บันทึกไว้ |
---|
การส่งคืน
- วัตถุที่สามารถใช้เพื่อเรียกฟังก์ชันได้
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | หากไม่พบ signatureKey ในโมเดลที่บันทึกไว้นี้ |
---|
โหลด SavedModelBundle สาธารณะแบบคงที่ (แท็ก string ExportDir, String...)
โหลดโมเดลที่บันทึกไว้จากไดเร็กทอรีการส่งออก ควรสร้างโมเดลที่กำลังโหลดโดยใช้ Saved Model API
วิธีนี้เป็นชวเลขสำหรับ:
SavedModelBundle.loader().withTags(tags).load();
พารามิเตอร์
ส่งออกDir | เส้นทางไดเร็กทอรีที่มีโมเดลที่บันทึกไว้ |
---|---|
แท็ก | แท็กที่ระบุ metagraphdef เฉพาะที่จะโหลด |
การส่งคืน
- กลุ่มที่มีกราฟและเซสชันที่เกี่ยวข้อง
ตัวโหลด SavedModelBundle.Loader สาธารณะแบบคงที่ (String ExportDir)
โหลดโมเดลที่บันทึกไว้
ส่งคืนออบเจ็กต์ Loader
ที่สามารถตั้งค่าตัวเลือกการกำหนดค่าก่อนที่จะโหลดโมเดลจริง
พารามิเตอร์
ส่งออกDir | เส้นทางไดเร็กทอรีที่มีโมเดลที่บันทึกไว้ |
---|
MetaGraphDef สาธารณะ metaGraphDef ()
ส่งกลับ บัฟเฟอร์โปรโตคอล MetaGraphDef ที่เกี่ยวข้องกับแบบจำลองที่บันทึกไว้