عملیاتی که پارامترهای بهینه سازی را در حافظه جاسازی شده بارگذاری می کند.
عملیاتی که پارامترهای بهینه سازی را در حافظه جاسازی شده بارگذاری می کند. باید قبل از آن یک عملیات ConfigureTPUEmbeddingHost باشد که پیکربندی صحیح جدول جاسازی را تنظیم می کند. به عنوان مثال، این عملیات برای نصب پارامترهایی استفاده می شود که قبل از اجرای یک حلقه آموزشی از یک نقطه بازرسی بارگذاری می شوند. برای Adagrad، auxiliary1 باید انباشته باشد. برای SGD، تمام مقادیر کمکی* باید خالی باشند. برای FTRL، auxiliary1 باید انباشتگر و auxiliary2 باید عبارت خطی باشد. برای ADAM، auxiliary1 باید لحظه و auxiliary2 باید سرعت باشد.
روش های عمومی
استاتیک LoadAllTPUEmbeddingParameters | ایجاد ( Scope scope, Iterable< Operand <Float>> پارامترها, Iterable< Operand <Float>> auxiliary1, Iterable< Operand <Float>> auxiliary2, Iterable< Operand <Float>> auxiliary3, Iterable< Operand <Float>> auxiliary4, Iterable< Operand <Float>> auxiliary5, Iterable< Operand <Float>> auxiliary6, Iterable< Operand <Float>> auxiliary7, String config, Long numShards, Long shardId) روش کارخانه برای ایجاد کلاسی که یک عملیات LoadAllTPUEmbeddingParameters جدید را بسته بندی می کند. |
روش های ارثی
روش های عمومی
عمومی استاتیک LoadAllTPUEmbeddingParameters ایجاد ( دامنه دامنه ، تکرارپذیر< Operand <Float>> پارامترها ، تکرارپذیر< Operand <Float>> auxiliary1، تکرارپذیر< Operand <Float>> auxiliary2، تکرارپذیر< Operand <Float>> کمکی3، Iterable > auxiliary4, Iterable< Operand <Float>> auxiliary5, Iterable< Operand <Float>> auxiliary6, Iterable< Operand <Float>> auxiliary7, String config, Long numShards, Long shardId)
روش کارخانه برای ایجاد کلاسی که یک عملیات LoadAllTPUEmbeddingParameters جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
مولفه های | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی پارامترهای اولیه جدول جاسازی برای استفاده در جستجوهای جاسازی. |
کمکی 1 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه اولین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی برای آن دسته از جداول که الگوریتم های بهینه سازی آنها حداقل یک پارامتر کمکی ندارند نادیده گرفته می شود (و بنابراین می تواند خالی باشد). |
کمکی 2 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه دومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جدول هایی که الگوریتم های بهینه سازی آنها حداقل دو مورد کمکی ندارند. |
کمکی 3 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه سومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جداول که الگوریتم های بهینه سازی آنها دارای سه پارامتر کمکی نیستند. |
کمکی 4 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه دومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جدول هایی که الگوریتم های بهینه سازی آنها حداقل چهار مورد کمکی ندارند. |
کمکی 5 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه سومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جداول که الگوریتم های بهینه سازی آنها دارای پنج پارامتر کمکی نیستند. |
کمکی 6 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه دومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جدول هایی که الگوریتم های بهینه سازی آنها حداقل شش مورد کمکی ندارند. |
کمکی 7 | فهرستی از تانسورها، یکی برای هر جدول جاسازی، حاوی مقادیر اولیه سومین پارامتر بهینهسازی کمکی برای استفاده در بهروزرسانیهای حلقه آموزشی. شکل هر ورودی نادیده گرفته می شود (و بنابراین می تواند خالی باشد) برای آن دسته از جداول که الگوریتم های بهینه سازی آنها پارامترهای کمکی sevan را ندارند. |
پیکربندی | یک پروتوی TPUEmbeddingConfiguration که پارامترهای جدول در حال بارگذاری را توصیف می کند و به صورت سریال به رشته ای تبدیل می شود. |
numShards | تعداد خرده هایی که جداول تعبیه شده به آنها تقسیم می شوند. |
shardId | شناسه خرده برای این عملیات. |
برمی گرداند
- یک نمونه جدید از LoadAllTPUEmbeddingParameters