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