black formatted files before changes

This commit is contained in:
Jan Kowalczyk
2024-06-28 11:36:46 +02:00
parent d33c6b1e16
commit 71f9662022
40 changed files with 2938 additions and 1260 deletions

View File

@@ -6,8 +6,17 @@ from .base_net import BaseNet
class BaseTrainer(ABC):
"""Trainer base class."""
def __init__(self, optimizer_name: str, lr: float, n_epochs: int, lr_milestones: tuple, batch_size: int,
weight_decay: float, device: str, n_jobs_dataloader: int):
def __init__(
self,
optimizer_name: str,
lr: float,
n_epochs: int,
lr_milestones: tuple,
batch_size: int,
weight_decay: float,
device: str,
n_jobs_dataloader: int,
):
super().__init__()
self.optimizer_name = optimizer_name
self.lr = lr