Add Benchmarking to PyTorch Code

Easyauto-graded

Profile PyTorch operations using torch.utils.benchmark, measuring GPU/CPU time and comparing implementations.

Solve it

Open in ColabQuestion notebookSolution

Check your answer

The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.

pip install torchleet

from torchleet import check
check("benchmark", SimpleNN)