Method forward

benchmarks/python/conv2d_train_bench_cpu.py:98–99

(self, input)

            )

        def forward(self, input):
            return self.net(input)

    benchNet = BenchNetTorch(3).to(device)
    optim = torch.optim.Adam(benchNet.parameters(), lr=1e-3)