20 lines
476 B
Python
20 lines
476 B
Python
from abc import ABC, abstractmethod
|
|
|
|
import torch.nn as nn
|
|
from torch.utils.data import DataLoader
|
|
|
|
|
|
class Trainer(ABC):
|
|
"""Abstract class for trainers."""
|
|
|
|
@abstractmethod
|
|
def execute(
|
|
self,
|
|
model: nn.Module | type[nn.Module] | None,
|
|
context_length: int,
|
|
train_loader: DataLoader,
|
|
validation_loader: DataLoader,
|
|
n_epochs: int | None,
|
|
device: str
|
|
) -> nn.Module:
|
|
pass
|