elektronn3.training.triplettrainer module

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

Bases: elektronn3.training.trainer.Trainer

epoch: int
exp_name: str
out_channels: Optional[int]
save_path: str
step: int
tb: tensorboardX.SummaryWriter
terminate: bool
train_loader: torch.utils.data.DataLoader
valid_loader: torch.utils.data.DataLoader