SavedModelBundle

คลาสสาธารณะ SavedModelBundle

SavedModelBundle แสดงถึงโมเดลที่โหลดจากที่จัดเก็บข้อมูล

แบบจำลองประกอบด้วยคำอธิบายของการคำนวณ ( Graph ) Session ที่มีเทนเซอร์ (เช่น พารามิเตอร์หรือตัวแปรในกราฟ) กำหนดค่าเริ่มต้นเป็นค่าที่บันทึกไว้ในที่จัดเก็บ และคำอธิบายของแบบจำลองเป็น บัฟเฟอร์โปรโตคอล MetaGraphDef

คลาสที่ซ้อนกัน

ระดับ SavedModelBundle ผู้ส่งออก ตัวเลือกสำหรับการส่งออก SavedModel
ระดับ SavedModelBundle.Loader ตัวเลือกสำหรับการโหลด SavedModel

ค่าคงที่

สตริง DEFAULT_TAG

วิธีการสาธารณะ

แผนที่<สตริง เทนเซอร์ >
โทร (แผนที่ <สตริง, เทนเซอร์ > อาร์กิวเมนต์)
เรียกใช้ฟังก์ชันเริ่มต้นโดยตรงจากโมเดลนี้
เป็นโมฆะ
ปิด ()
เผยแพร่ทรัพยากร ( Graph และ Session ) ที่เกี่ยวข้องกับบันเดิลโมเดลที่บันทึกไว้
SavedModelBundle.Exporter แบบคงที่
ผู้ส่งออก (สตริงexportDir)
ส่งออกโมเดลที่บันทึกไว้
คอนกรีตฟังก์ชั่น
ฟังก์ชั่ น (String SignatureKey)
ส่งกลับ ConcreteFunction ที่สอดคล้องกับลายเซ็นของฟังก์ชัน
กราฟ
กราฟ ()
ส่งกลับกราฟที่อธิบายการคำนวณที่ทำโดยแบบจำลอง
SavedModelBundle แบบคงที่
โหลด (สตริง ExportDir, สตริง... แท็ก)
โหลดโมเดลที่บันทึกไว้จากไดเร็กทอรีการส่งออก
SavedModelBundle.Loader แบบคงที่
ตัวโหลด (สตริงส่งออกDir)
โหลดโมเดลที่บันทึกไว้
MetaGraphDef
เมตากราฟDef ()
ส่งกลับ บัฟเฟอร์โปรโตคอล MetaGraphDef ที่เกี่ยวข้องกับแบบจำลองที่บันทึกไว้
การประชุม
การประชุม ()
ส่งคืน Session ที่จะดำเนินการคำนวณโดยใช้โมเดล
รายการ< ลายเซ็น >
ลายเซ็น ()
ส่งกลับลายเซ็นของฟังก์ชันทั้งหมดที่มีอยู่ในโมเดลที่บันทึกไว้นี้

วิธีการสืบทอด

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ DEFAULT_TAG

ค่าคงที่: "เสิร์ฟ"

วิธีการสาธารณะ

แผนที่สาธารณะ <สตริง, เทนเซอร์ > โทร (อาร์กิวเมนต์แผนที่ <สตริง, เทนเซอร์ >)

เรียกใช้ฟังก์ชันเริ่มต้นโดยตรงจากโมเดลนี้

การเลือกฟังก์ชันเริ่มต้นจะเสร็จสิ้นตามเงื่อนไขแรกที่เป็นจริงต่อไปนี้:

  • ฟังก์ชันนี้เป็นลายเซ็นเดียวที่แนบกับกราฟหลักของแบบจำลองที่บันทึกไว้นี้
  • ฟังก์ชันนี้ถูกแมปกับชื่อลายเซ็นเริ่มต้น ซึ่งก็คือ "serving_default"

ผู้เรียกมีหน้าที่ปิดเทนเซอร์ที่ส่งคืนทั้งหมด

พารามิเตอร์
ข้อโต้แย้ง รายการเทนเซอร์อินพุต แมปตามชื่อลายเซ็น
การส่งคืน
  • รายการเทนเซอร์เอาท์พุต แมปตามชื่อลายเซ็น
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย หากไม่มีฟังก์ชันใดสามารถเลือกได้ตามค่าเริ่มต้น

โมฆะสาธารณะ ปิด ()

เผยแพร่ทรัพยากร ( Graph และ Session ) ที่เกี่ยวข้องกับบันเดิลโมเดลที่บันทึกไว้

ผู้ส่งออก 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 ที่เกี่ยวข้องกับแบบจำลองที่บันทึกไว้

เซ สชั่น สาธารณะ ()

ส่งคืน Session ที่จะดำเนินการคำนวณโดยใช้โมเดล

การส่งคืน
  • เซสชันเริ่มต้น

รายการสาธารณะ< ลายเซ็น > ลายเซ็น ()

ส่งกลับลายเซ็นของฟังก์ชันทั้งหมดที่มีอยู่ในโมเดลที่บันทึกไว้นี้