หน้าแรก>การเขียนโปรแกรมที่เกี่ยวข้อง>หลาม
ดาวน์โหลด

เบรวิต้า

pytestตัวอย่าง pytest

Brevitas เป็นห้องสมุด Pytorch สำหรับการหาปริมาณเครือข่ายประสาทโดยสนับสนุนทั้ง การฝึกอบรมหลังการฝึกอบรม (PTQ) และ การฝึกอบรมเชิงปริมาณ (QAT)

โปรดทราบว่า Brevitas เป็นโครงการวิจัยและไม่ใช่ผลิตภัณฑ์ Xilinx อย่างเป็นทางการ

หากคุณชอบโครงการนี้โปรดพิจารณา repo นี้เนื่องจากเป็นวิธีที่ง่ายที่สุดและดีที่สุดในการสนับสนุน

ความต้องการ

  • Python> = 3.8
  • Pytorch> = 1.9.1, <= 2.1 (เวอร์ชันล่าสุดจะยังไม่ทดลอง)
  • Windows, Linux หรือ MacOS
  • การเร่งความเร็วเวลาการฝึกอบรม GPU ( ไม่บังคับ แต่แนะนำ)

การติดตั้ง

คุณสามารถติดตั้งรีลีสล่าสุดจาก PYPI: 

pip install brevitas

เริ่มต้น

ปัจจุบัน Brevitas เสนอ QuantLSTM ใช้งานเชิงปริมาณของชั้น pytorch ที่พบมากที่สุด QuantConvTranspose2d ใช้ใน DNN ภาย QuantRNN brevitas.nn เช่น QuantConv1d , QuantConv2d , QuantConvTranspose1d , QuantMultiheadAttention สำหรับแต่ละเลเยอร์เหล่านี้การหาปริมาณของเทนเซอร์ที่แตกต่างกัน (อินพุต, น้ำหนัก, อคติ, เอาต์พุต ฯลฯ ) สามารถปรับแต่งเป็นรายบุคคลตามการตั้งค่าปริมาณที่หลากหลาย

เป็นข้อมูลอ้างอิงสำหรับ PTQ, Brevitas จัดเตรียมตัวอย่างการไหลของผู้ใช้สำหรับโมเดลการจำแนกประเภท Imagenet ภายใต้ brevitas_examples.imagenet_classification.ptq ที่หาปริมาณแบบจำลอง Torchvision อินพุตโดยใช้ PTQ ภายใต้การกำหนดค่าปริมาณที่แตกต่างกัน

สำหรับข้อมูลเพิ่มเติมเช็คเอาต์คู่มือการเริ่มต้นใช้งานของเรา

อ้างถึง

หากคุณนำ Brevitas มาใช้โปรดอ้างอิงเป็น: 

 @software{brevitas,
  author       = {Alessandro Pappalardo},
  title        = {Xilinx/brevitas},
  year         = {2023},
  publisher    = {Zenodo},
  doi          = {10.5281/zenodo.3333552},
  url          = {https://doi.org/10.5281/zenodo.3333552}
}

ประวัติศาสตร์

  • 2024/10/10 - รุ่นรุ่น 0.11.0 ดูบันทึกย่อ
  • 2024/07/23 - รุ่นรองรุ่น 0.10.3 ดูบันทึกย่อการเปิดตัว
  • 2024/02/19 - รุ่นรองรุ่น 0.10.2 ดูบันทึกย่อการเปิดตัว
  • 2024/02/15 - รุ่นรองรุ่น 0.10.1 ดูบันทึกย่อการเปิดตัว
  • 2023/12/08 - รุ่นรุ่น 0.10.0 ดูบันทึกย่อ
  • 2023/04/28 - รุ่นรองรุ่น 0.9.1 ดูบันทึกย่อการเปิดตัว
  • 2023/04/21 - รุ่นรุ่น 0.9.0 ดูบันทึกย่อการเปิดตัว
  • 2023/01/10 - รุ่นรุ่น 0.8.0 ดูบันทึกย่อ
  • 2021/12/13 - รุ่นรุ่น 0.7.1 แก้ไขปัญหามากมาย เพิ่มสมุดบันทึกการสอน TVMCON 2021
  • 2021/11/03 - รีลีสเวอร์ชั่น 0.7.0 (สร้าง 1) บน PYPI เพื่อแก้ไขปัญหาบรรจุภัณฑ์
  • 2021/10/29 - รุ่นรุ่น 0.7.0 ดูบันทึกย่อ
  • 2021/06/04 - รุ่นรุ่น 0.6.0 ดูบันทึกการเปิดตัว
  • 2021/05/24 - ปล่อยเวอร์ชัน 0.5.1 แก้ไขปัญหาเล็กน้อย ดูบันทึกย่อ
  • 2021/05/06 - รุ่นรีลีส 0.5.0 ดูบันทึกย่อการเปิดตัว
  • 2021/03/15 - รุ่นรุ่น 0.4.0 เพิ่มการสนับสนุนสำหรับ __TORCH_FUNCTION__ ถึง Quanttensor
  • 2021/03/04 - รุ่นรุ่น 0.3.1, แก้ไขข้อผิดพลาดพร้อมการเริ่มต้นการเริ่มต้นจากสถิติที่ w/ ignore_missing_keys = 1
  • 2021/03/01 - รุ่นรุ่น 0.3.0 ใช้งาน enum และตัวแก้รูปร่างภายในหัวฉีดพึ่งพาขยาย สิ่งนี้ช่วยให้ปริมาณการประกาศตัวเองมีอยู่ในตัวเอง
  • 2021/02/04 - รุ่นรุ่น 0.2.1 รวมถึงการแก้ไขข้อผิดพลาดต่าง ๆ ของ Quanttensor w/zero -point
  • 2021/01/30 - รุ่นแรกรุ่น 0.2.0 บน PYPI
ขยาย
ข้อมูลเพิ่มเติม
แอปที่เกี่ยวข้อง
แนะนำสำหรับคุณ
ข้อมูลที่เกี่ยวข้อง ทั้งหมด