doctor and model test
This commit is contained in:
11
Project_Model/Libs/Batch/Classes/BatchEmbedder.py
Normal file
11
Project_Model/Libs/Batch/Classes/BatchEmbedder.py
Normal file
@@ -0,0 +1,11 @@
|
||||
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, )
|
||||
Reference in New Issue
Block a user