Modernize to use ModuleList, Fix some types

2026-05-22 23:59:40 +02:00 · 2025-02-12 17:06:20 +01:00
parent dfe7ab3b6f
commit c6d16c0cf6
1 changed files with 0 additions and 1 deletions
@@ -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]