SavedModelBundle แสดงถึงโมเดลที่โหลดจากที่เก็บข้อมูล
รุ่นประกอบด้วยรายละเอียดของการคำนวณ (เป็น Graph
) ซึ่งเป็น Session
กับเทนเซอร์ (เช่นพารามิเตอร์หรือตัวแปรในกราฟ) เริ่มต้นได้ที่ค่าบันทึกไว้ในการจัดเก็บข้อมูลและรายละเอียดของรูปแบบ (เป็นตัวแทนต่อเนื่องของการเป็น บัฟเฟอร์โปรโตคอล MetaGraphDef ).
คลาสที่ซ้อนกัน
ระดับ | ที่บันทึกไว้ ModelBundle.Loader | ตัวเลือกสำหรับการโหลด SavedModel |
วิธีการสาธารณะ
โมฆะ | |
กราฟ | กราฟ () ส่งกลับกราฟที่อธิบายการคำนวณที่ดำเนินการโดยโมเดล |
คง SavedModelBundle | โหลด (String exportDir สตริง ... แท็ก) โหลดโมเดลที่บันทึกไว้จากไดเร็กทอรีเอ็กซ์พอร์ต |
คง SavedModelBundle.Loader | รถตักดิน (String exportDir) โหลดโมเดลที่บันทึกไว้ |
ไบต์[] | metaGraphDef () ผลตอบแทนต่อเนื่อง MetaGraphDef บัฟเฟอร์โปรโตคอล ที่เกี่ยวข้องกับรูปแบบการบันทึกไว้ |
การประชุม |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง SavedModelBundle โหลด (String exportDir สตริง ... แท็ก)
โหลดโมเดลที่บันทึกไว้จากไดเร็กทอรีเอ็กซ์พอร์ต รูปแบบที่จะถูกโหลดควรได้รับการสร้างขึ้นโดยใช้ ที่บันทึกไว้รุ่น API
วิธีนี้เป็นวิธีจดชวเลขสำหรับ:
SavedModelBundle.loader().withTags(tags).load();
พารามิเตอร์
ส่งออกDir | เส้นทางไดเรกทอรีที่มีโมเดลที่บันทึกไว้ |
---|---|
แท็ก | แท็กที่ระบุ metagraphdef เฉพาะที่จะโหลด |
คืนสินค้า
- กลุ่มที่มีกราฟและเซสชันที่เกี่ยวข้อง
สาธารณะคง SavedModelBundle.Loader รถตักดิน (String exportDir)
โหลดโมเดลที่บันทึกไว้
ผลตอบแทนLoader
วัตถุที่สามารถตั้งค่าตัวเลือกการกำหนดค่าก่อนที่จริงโหลดรูปแบบ พารามิเตอร์
ส่งออกDir | เส้นทางไดเรกทอรีที่มีโมเดลที่บันทึกไว้ |
---|
สาธารณะ byte [] metaGraphDef ()
ผลตอบแทนต่อเนื่อง MetaGraphDef บัฟเฟอร์โปรโตคอล ที่เกี่ยวข้องกับรูปแบบการบันทึกไว้