LoadAllTPUEmbeddingParameters

सार्वजनिक अंतिम वर्ग LoadAllTPUEmbeddingParameters

एक ऑप जो अनुकूलन मापदंडों को एम्बेडिंग मेमोरी में लोड करता है।

एक ऑप जो अनुकूलन मापदंडों को एम्बेडिंग मेमोरी में लोड करता है। कॉन्फिगरटीपीयूएम्बेडिंगहोस्ट ऑप से पहले होना चाहिए जो सही एम्बेडिंग टेबल कॉन्फ़िगरेशन सेट करता है। उदाहरण के लिए, इस ऑप का उपयोग उन मापदंडों को स्थापित करने के लिए किया जाता है जो एक प्रशिक्षण लूप निष्पादित होने से पहले चेकपॉइंट से लोड किए जाते हैं। एडाग्राड के लिए, सहायक1 संचायक होना चाहिए। एसजीडी के लिए, सभी सहायक* मान खाली होने चाहिए। एफटीआरएल के लिए, सहायक1 संचायक होना चाहिए और सहायक2 रैखिक पद होना चाहिए। ADAM के लिए, सहायक1 को संवेग होना चाहिए और सहायक2 को वेग होना चाहिए।

सार्वजनिक तरीके

स्थिर LoadAllTPUEmbeddingParameters
बनाएं ( स्कोप स्कोप, इटरेबल< ऑपरेंड <फ्लोट>> पैरामीटर्स, इटरेबल< ऑपरेंड <फ्लोट>> सहायक1, इटरेबल< ऑपरेंड <फ्लोट>> सहायक2, इटरेबल< ऑपरेंड <फ्लोट>> सहायक3, इटरेबल< ऑपरेंड <फ्लोट>> सहायक4, पुनरावर्तनीय< ऑपरेंड <फ्लोट>> सहायक5, पुनरावर्तनीय< संचालन <फ्लोट>> सहायक6, पुनरावर्तनीय< संकार्य <फ्लोट>> सहायक7, स्ट्रिंग कॉन्फिगरेशन, लंबी संख्याशार्ड्स, लंबी शार्डआईडी)
एक नए LoadAllTPUEmbeddingParameters ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

विरासत में मिली विधियाँ

सार्वजनिक तरीके

सार्वजनिक स्थैतिक LoadAllTPUEmbeddingParameters बनाते हैं ( स्कोप स्कोप, Iterable< ऑपरेंड <फ्लोट>> पैरामीटर, Iterable< ऑपरेंड <फ्लोट>> सहायक1, Iterable< ऑपरेंड <फ्लोट>> सहायक2, Iterable< ऑपरेंड <फ्लोट>> सहायक3, Iterable< ऑपरेंड <फ्लोट> > सहायक4, पुनरावर्तनीय < ऑपरेंड <फ्लोट>> सहायक5, पुनरावर्तनीय< संचालन <फ्लोट>> सहायक6, पुनरावर्तनीय< संचालन <फ्लोट>> सहायक7, स्ट्रिंग कॉन्फिगरेशन, लंबी संख्याशार्ड्स, लंबी शार्डआईडी)

एक नए LoadAllTPUEmbeddingParameters ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

पैरामीटर
दायरा वर्तमान दायरा
पैरामीटर टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग लुकअप में उपयोग करने के लिए प्रारंभिक एम्बेडिंग तालिका पैरामीटर शामिल हैं।
सहायक1 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए पहले सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को अनदेखा किया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में कम से कम एक सहायक पैरामीटर नहीं होता है।
सहायक2 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए दूसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को अनदेखा किया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में कम से कम दो सहायक नहीं होते हैं
सहायक3 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए तीसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को अनदेखा किया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में तीन सहायक पैरामीटर नहीं होते हैं।
सहायक4 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए दूसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को नजरअंदाज कर दिया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में कम से कम चार सहायक नहीं होते हैं
सहायक5 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए तीसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को अनदेखा किया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में पांच सहायक पैरामीटर नहीं होते हैं।
सहायक6 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए दूसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को नजरअंदाज कर दिया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में कम से कम छह सहायक नहीं होते हैं
सहायक7 टेंसरों की एक सूची, प्रत्येक एम्बेडिंग तालिका के लिए एक, जिसमें एम्बेडिंग प्रशिक्षण लूप अपडेट में उपयोग करने के लिए तीसरे सहायक अनुकूलन पैरामीटर के प्रारंभिक मान शामिल हैं। उन तालिकाओं के लिए प्रत्येक प्रविष्टि के आकार को अनदेखा किया जाता है (और इस प्रकार खाली हो सकता है) जिनके अनुकूलन एल्गोरिदम में सात सहायक पैरामीटर नहीं होते हैं।
कॉन्फ़िग एक TPUEmbeddingConfiguration प्रोटो लोड किए जा रहे तालिका मापदंडों का वर्णन करता है, जो एक स्ट्रिंग में क्रमबद्ध है।
numShards उन शार्डों की संख्या जिनमें एम्बेडिंग तालिकाओं को विभाजित किया गया है।
शार्डआई.डी इस ऑपरेशन के लिए शार्ड का पहचानकर्ता.
रिटर्न
  • LoadAllTPUEmbeddingParameters का एक नया उदाहरण