Modernize to use ModuleList, Fix some types

This commit is contained in:
henryruhs committed 2025-03-11 14:43:09 +01:00
1 parent dfe7ab3b6f
commit c6d16c0cf6
1 file changed
-1
-1
View File
@@ -2,7 +2,6 @@ from typing import Any, Tuple
from numpy.typing import NDArray
from torch import Tensor
from torch.utils.data import DataLoader
Embedding = NDArray[Any]
EmbeddingDataset = NDArray[Embedding]