Trainer

Module: generative_models.training.trainer

Source: generative_models/training/trainer.py

Overview

Trainer for generative models.

Classes

Trainer

class Trainer

TrainingState

class TrainingState

Functions

init

def __init__()

create

def create()

evaluate

def evaluate()

generate_samples

def generate_samples()

load_checkpoint

def load_checkpoint()

loss_fn

def loss_fn()

save_checkpoint

def save_checkpoint()

train

def train()

train_epoch

def train_epoch()

train_step

def train_step()

validate_step

def validate_step()

Module Statistics

• Classes: 2
• Functions: 11
• Imports: 10