Function create_model

intermediate_source/intermediate_data_loading_tutorial.py:162–164

Create a conv+transformer model for benchmarking.

()

def create_model():
    """Create a conv+transformer model for benchmarking."""
    return SmallTransformerModel().to(device)


def train_and_benchmark(loader, max_batches=160, epochs=10, prefetch_device=None):