The benchmark classes allow us to measure the peak memory usage and required time for both inference and training. | |
Hereby, inference is defined by a single forward pass, and training is defined by a single forward pass and | |
backward pass. | |
The benchmark classes [PyTorchBenchmark] and [TensorFlowBenchmark] expect an object of type [PyTorchBenchmarkArguments] and | |
[TensorFlowBenchmarkArguments], respectively, for instantiation. |