MyModel1 Class — pytorch Architecture

Architecture documentation for the MyModel1 class in microbench.py from the pytorch codebase.

Class python

Entity Profile

Source Code

benchmarks/dynamo/microbenchmarks/microbench.py lines 55–65

class MyModel1(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.model = torch.nn.Sequential(
            torch.nn.Linear(1024, 1024),
            torch.nn.ReLU(),
        )

    def forward(self, input):
        # return (self.model(input) + 1,)
        return (self.model(input),)

Source

View on GitHub

Analyze Your Own Codebase

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

Try Supermodel Free