11 lines
347 B
Python
11 lines
347 B
Python
from ....Libs.Embedder.Classes.NanoSocratesEmbedder import NanoSocratesEmbedder
|
|
import torch
|
|
|
|
class BatchEmbedder(torch.nn.Module):
|
|
|
|
def __init__(self, vocabulary_size: int, embedding_size: int) -> None:
|
|
super().__init__()
|
|
self.__embedder = NanoSocratesEmbedder(vocabulary_size,embedding_size)
|
|
|
|
|
|
def forward(self, ) |