Home / Class/ Benchmark Class — pytorch Architecture

Benchmark Class — pytorch Architecture

Architecture documentation for the Benchmark class in benchmark.py from the pytorch codebase.

Class python

Entity Profile

Source Code

benchmarks/distributed/ddp/benchmark.py lines 155–172

class Benchmark:
    def __init__(self, device, distributed_backend, bucket_size):
        self.device = device
        self.batch_size = 32
        self.distributed_backend = distributed_backend
        self.bucket_size = bucket_size

    def __str__(self):
        raise NotImplementedError

    def create_model(self):
        raise NotImplementedError

    def generate_inputs(self):
        raise NotImplementedError

    def generate_target(self):
        raise NotImplementedError

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free