الصفحة الرئيسية>المتعلقة بالبرمجة>شفرة المصدر الأخرى
تنزيل

Sten: واجهة للتفاؤل الفعال في Pytorch

يهدف Sten إلى حل الأسئلة التالية التي ظلت دون إجابة في التنفيذ الحالي (Torch.sparse) من Sparsity في Pytorch 1.11.

  • كيفية دمج استراتيجية تفوق في النموذج واستخدامها في وقت التشغيل؟
  • كيف تحافظ على مستوى sparsity نفسه أثناء التدريب؟
  • كيفية تمكين دعم Autograd الكامل؟
  • كيفية تمكين التنسيقات المتفرقة المخصصة وتطبيقات المشغل؟

أمثلة

  • تحقق من build_from_scratch.ipynb لمعرفة مثال استخدام الواجهة لإنشاء وحدة Pytorch من نقطة الصفر.
  • تحقق من modify_existing.ipynb لمعرفة مثال تحويل وحدة pytorch الكثيفة الموجودة إلى متفرقة.
  • تحقق من custom_implementations.ipynb لمعرفة مثال تسجيل التطبيقات المخصصة للفروسية والمشغلين التي تتطابق مع تنسيقات محددة من موترات الإدخال والمخرجات.

بداية سريعة 

 git clone https://github.com/spcl/sten.git
cd sten
python -m venv venv
source venv/bin/activate
pip install .
python tests/test_api.py

تثبيت 

 pip install sten

تنظيم الكود

يقع التنفيذ الأساسي في Sten.py. تقع أمثلة دفتر جوبتر في دليل أمثلة. يمكن العثور على المزيد من الأمثلة في شكل اختبارات في دليل الاختبارات. يمكن إجراء الاختبارات عن طريق استدعاء pytest في جذر المشروع.

يوسع
معلومات إضافية
تطبيقات ذات صلة
نوصي لك
أخبار ذات صلة الكل