דף זה תורגם על ידי Cloud Translation API.
Switch to English

הורדות מודל מטמון מה- TF Hub

סיכום

הספרייה של tensorflow_hub במטמון מודלים במערכת הקבצים כאשר הם הורדו מ- tfhub.dev (או מאתרי אירוח אחרים) והורדו לדחיסה. מיקום ההורדה TFHUB_CACHE_DIR כברירת מחדל לספרייה זמנית מקומית אך ניתן להתאים אותו על ידי הגדרת משתנה הסביבה TFHUB_CACHE_DIR (מומלץ) או העברת דגל שורת הפקודה - --tfhub_cache_dir . בעת שימוש במיקום מתמשך, שים לב כי אין ניקוי אוטומטי.

הקריאות לפונקציות tensorflow_hub בקוד הפייתון בפועל יכולות וצריכות להמשיך להשתמש בכתובות אתרים tfhub.dev של דגמים, הניידים על פני מערכות וניתנים לניווט לתיעוד.

סביבות ביצוע ספציפיות

אם וכיצד צריך לשנות את ברירת המחדל של TFHUB_CACHE_DIR תלוי בסביבת הביצוע.

פועל במקום בתחנת עבודה

עבור משתמשים שמריצים תוכניות TensorFlow בתחנת העבודה שלהם, זה צריך לעבוד רק ברוב המקרים כדי להמשיך ולהשתמש במיקום ברירת המחדל /tmp/tfhub_modules , או מה שזה לא יהיה שפייתון מחזיר עבור os.path.join(tempfile.gettempdir(), "tfhub_modules") .

משתמשים המעדיפים מטמון קבוע על פני אתחול מחדש של המערכת יכולים במקום זאת להגדיר את TFHUB_CACHE_DIR למיקום בספריית הבית שלהם. לדוגמא, משתמש במעטפת bash במערכת לינוקס יכול להוסיף שורה כמו הבאה ל- ~/.bashrc

export TFHUB_CACHE_DIR=$HOME/.cache/tfhub_modules

... הפעל מחדש את הקליפה ואז ישמש במיקום זה.

פועל על TPU במחברות של Colab

להפעלת TensorFlow על מעבד ו- GPU מתוך מחברת Colab , שימוש במיקום המטמון המקומי המוגדר כברירת מחדל פשוט צריך לעבוד.

פועל על נציגי TPU למכונה אחרת שאין לה גישה למיקום המטמון המקומי המוגדר כברירת מחדל. משתמשים עם דלי אחסון ענן של Google (GCS) משלהם יכולים לעקוף זאת על ידי הגדרת ספרייה בדלי זה כמיקום המטמון עם קוד כמו

import os
os.environ["TFHUB_CACHE_DIR"] = "gs://my-bucket/tfhub-modules-cache"

... לפני tensorflow_hub לספריית tensorflow_hub .