تنزيل prototypical networks - تنزيل رمز المصدر prototypical networks

prototypical networks

بايثون

1.0.0

تنزيل

الشبكات النموذجية للتعلم قليلًا

رمز لشبكات NIPS 2017 Paper Orgeical لتعلم القليل من اللقطة.

إذا كنت تستخدم هذا الرمز ، يرجى الاستشهاد بالورقة:

 @inproceedings{snell2017prototypical,
  title={Prototypical Networks for Few-shot Learning},
  author={Snell, Jake and Swersky, Kevin and Zemel, Richard},
  booktitle={Advances in Neural Information Processing Systems},
  year={2017}
 }

تدريب شبكة أولية

تثبيت التبعيات

تم اختبار هذا الرمز على Ubuntu 16.04 مع Python 3.6 و Pytorch 0.4.
تثبيت pytorch و torchvision.
تثبيت Torchnet عن طريق تشغيل pip install git+https://github.com/pytorch/tnt.git@master .
قم بتثبيت حزمة Protonets عن طريق تشغيل python setup.py install أو python setup.py develop .

قم بإعداد مجموعة بيانات Omniglot

تشغيل sh download_omniglot.sh .

تدريب النموذج

قم بتشغيل python scripts/train/few_shot/run_train.py . سيؤدي ذلك إلى تشغيل التدريب ووضع النتائج في results .
- يمكنك تحديد دليل إخراج مختلف عن طريق تمرير الخيار --log.exp_dir EXP_DIR ، حيث يكون EXP_DIR دليل الإخراج المطلوب.
- إذا كنت تعمل على وحدة معالجة الرسومات ، فيمكنك تمرير الخيار --data.cuda .
إعادة تشغيل في Trainval Mode python scripts/train/few_shot/run_trainval.py . سيؤدي هذا إلى حفظ النموذج الخاص بك في results/trainval افتراضيًا.