elektronn3.training.triplettrainer module

class elektronn3.training.triplettrainer.TripletTrainer(*args, **kwargs)[source]

Bases: Trainer

epoch: int
exp_name: str
out_channels: Optional[int]
save_path: str
step: int
tb: SummaryWriter
terminate: bool
train_loader: DataLoader
valid_loader: DataLoader