Total time Total avg Total min Total max Total std Self time Self avg Self min Self max Self std Calls Size Function ========== ========== ========== ========== ========== ========== ========== ========== ========== ========== ========== ========== ==================== 10.510 s 10.510 s 10.510 s 10.510 s 462.518 us 462.518 us 462.518 us 462.518 us 1 1 __main__. 6.916 s 1.383 s 1.100 ms 3.945 s 2.147 s 6.916 s 1.383 s 1.100 ms 3.945 s 2.147 s 5 1 importlib._bootstrap._find_and_load 3.594 s 3.594 s 3.594 s 3.594 s 937.681 us 937.681 us 937.681 us 937.681 us 1 1 main 3.049 s 3.049 s 3.049 s 3.049 s 5.724 ms 5.724 ms 5.724 ms 5.724 ms 1 1 train 1.380 s 27.616 ms 156.163 us 163.812 ms 20.257 ms 1.380 s 27.616 ms 156.163 us 163.812 ms 20.257 ms 50 1 torch.utils.data.dataloader.__next__ 969.472 ms 20.627 ms 16.662 ms 49.245 ms 4.557 ms 969.472 ms 20.627 ms 16.662 ms 49.245 ms 4.557 ms 47 1 torch._tensor.backward 756.139 ms 15.752 ms 11.169 ms 122.808 ms 16.648 ms 1.531 ms 31.900 us 22.024 us 120.847 us 14.530 us 48 1 torch.nn.modules.module._wrapped_call_impl 754.608 ms 15.721 ms 11.144 ms 122.765 ms 16.645 ms 754.608 ms 15.721 ms 11.144 ms 122.765 ms 16.645 ms 48 1 forward 289.418 ms 289.418 ms 289.418 ms 289.418 ms 214.464 us 214.464 us 214.464 us 214.464 us 1 1 test 232.819 ms 116.409 ms 30.359 ms 202.459 ms 86.049 ms 232.819 ms 116.409 ms 30.359 ms 202.459 ms 86.049 ms 2 1 torchvision.datasets.mnist.__init__ 185.834 ms 3.953 ms 2.956 ms 5.670 ms 481.477 us 185.834 ms 3.953 ms 2.956 ms 5.670 ms 481.477 us 47 1 torch.optim.lr_scheduler.wrapper 33.386 ms 710.345 us 505.448 us 1.183 ms 176.899 us 33.386 ms 710.345 us 505.448 us 1.183 ms 176.899 us 47 1 torch._compile.inner 8.920 ms 4.460 ms 142.732 us 8.777 ms 4.317 ms 102.238 us 51.119 us 30.223 us 72.015 us 20.896 us 2 1 __init__ 5.031 ms 2.515 ms 345.772 us 4.685 ms 2.170 ms 5.031 ms 2.515 ms 345.772 us 4.685 ms 2.170 ms 2 1 torch.nn.modules.linear.__init__ 4.240 ms 4.240 ms 4.240 ms 4.240 ms 4.240 ms 4.240 ms 4.240 ms 4.240 ms 1 1 torch._dynamo.eval_frame._fn 3.272 ms 1.636 ms 618.672 us 2.653 ms 1.017 ms 3.272 ms 1.636 ms 618.672 us 2.653 ms 1.017 ms 2 1 torch.nn.modules.conv.__init__ 3.162 ms 3.162 ms 3.162 ms 3.162 ms 3.162 ms 3.162 ms 3.162 ms 3.162 ms 1 1 argparse.__init__ 2.969 ms 61.871 us 31.352 us 900.946 us 122.576 us 2.969 ms 61.871 us 31.352 us 900.946 us 122.576 us 48 1 torch.nn.functional.nll_loss 1.377 ms 125.185 us 105.062 us 203.094 us 27.680 us 1.377 ms 125.185 us 105.062 us 203.094 us 27.680 us 11 1 argparse.add_argument 1.104 ms 552.231 us 208.370 us 896.093 us 343.862 us 1.104 ms 552.231 us 208.370 us 896.093 us 343.862 us 2 1 torch.utils.data.dataloader.__iter__ 991.067 us 991.067 us 991.067 us 991.067 us 991.067 us 991.067 us 991.067 us 991.067 us 1 1 torch.optim.adadelta.__init__ 962.791 us 962.791 us 962.791 us 962.791 us 962.791 us 962.791 us 962.791 us 962.791 us 1 1 torch.nn.modules.module.to 766.344 us 38.317 us 19.431 us 97.956 us 18.682 us 766.344 us 38.317 us 19.431 us 97.956 us 18.682 us 20 1 builtins.len 744.564 us 67.687 us 6.000 us 158.900 us 35.741 us 744.564 us 67.687 us 6.000 us 158.900 us 35.741 us 11 1 builtins.print 573.161 us 286.580 us 79.696 us 493.465 us 206.885 us 573.161 us 286.580 us 79.696 us 493.465 us 206.885 us 2 1 torch.utils.data.dataloader.__init__ 488.786 us 488.786 us 488.786 us 488.786 us 488.786 us 488.786 us 488.786 us 488.786 us 1 1 Tensor.argmax 472.317 us 472.317 us 472.317 us 472.317 us 472.317 us 472.317 us 472.317 us 472.317 us 1 1 torch.nn.modules.module.eval 462.953 us 462.953 us 462.953 us 462.953 us 462.953 us 462.953 us 462.953 us 462.953 us 1 1 Tensor.sum 249.911 us 249.911 us 249.911 us 249.911 us 249.911 us 249.911 us 249.911 us 249.911 us 1 1 torch.nn.modules.module.train 244.519 us 81.506 us 37.980 us 138.843 us 42.319 us 244.519 us 81.506 us 37.980 us 138.843 us 42.319 us 3 1 torchvision.transforms.transforms.__init__ 242.122 us 242.122 us 242.122 us 242.122 us 242.122 us 242.122 us 242.122 us 242.122 us 1 1 Tensor.eq 215.204 us 2.241 us 0.930 us 12.284 us 1.679 us 215.204 us 2.241 us 0.930 us 12.284 us 1.679 us 96 1 Tensor.to 208.324 us 104.162 us 58.217 us 150.107 us 45.945 us 208.324 us 104.162 us 58.217 us 150.107 us 45.945 us 2 1 torch.nn.modules.dropout.__init__ 172.849 us 28.808 us 17.957 us 48.423 us 12.189 us 172.849 us 28.808 us 17.957 us 48.423 us 12.189 us 6 1 torch.nn.modules.module.__setattr__ 164.961 us 164.961 us 164.961 us 164.961 us 164.961 us 164.961 us 164.961 us 164.961 us 1 1 argparse.parse_args 103.538 us 103.538 us 103.538 us 103.538 us 103.538 us 103.538 us 103.538 us 103.538 us 1 1 torch.optim.lr_scheduler.__init__ 84.710 us 12.101 us 11.322 us 13.182 us 0.723 us 84.710 us 12.101 us 11.322 us 13.182 us 0.723 us 7 1 Tensor.item 58.163 us 58.163 us 58.163 us 58.163 us 58.163 us 58.163 us 58.163 us 58.163 us 1 1 torch.optim.lr_scheduler.step 55.927 us 9.321 us 8.415 us 10.245 us 0.608 us 55.927 us 9.321 us 8.415 us 10.245 us 0.608 us 6 1 str.format 51.498 us 51.498 us 51.498 us 51.498 us 13.170 us 13.170 us 13.170 us 13.170 us 1 1 __setattr__ 47.211 us 23.605 us 9.203 us 38.008 us 14.403 us 47.211 us 23.605 us 9.203 us 38.008 us 14.403 us 2 1 importlib._bootstrap._handle_fromlist 46.121 us 23.060 us 21.229 us 24.892 us 1.837 us 38.121 us 19.060 us 18.139 us 19.982 us 0.931 us 2 1 builtins.__build_class__ 34.179 us 34.179 us 34.179 us 34.179 us 34.179 us 34.179 us 34.179 us 34.179 us 1 1 Tensor.view_as 31.695 us 31.695 us 31.695 us 31.695 us 31.695 us 31.695 us 31.695 us 31.695 us 1 1 torch.backends.mps.is_available 24.748 us 12.374 us 10.100 us 14.648 us 2.274 us 24.748 us 12.374 us 10.100 us 14.648 us 2.274 us 2 1 torch.utils.data.sampler.__init__ 23.955 us 23.955 us 23.955 us 23.955 us 23.955 us 23.955 us 23.955 us 23.955 us 1 1 torch.autograd.grad_mode.__exit__ 21.021 us 21.021 us 21.021 us 21.021 us 21.021 us 21.021 us 21.021 us 21.021 us 1 1 torch.autograd.grad_mode.__enter__ 20.267 us 20.267 us 20.267 us 20.267 us 20.267 us 20.267 us 20.267 us 20.267 us 1 1 torch.nn.modules.module.__init__ 17.834 us 17.834 us 17.834 us 17.834 us 17.834 us 17.834 us 17.834 us 17.834 us 1 1 torch.cuda.is_available 9.628 us 9.628 us 9.628 us 9.628 us 8.016 us 8.016 us 8.016 us 8.016 us 1 1 __radd__ 9.173 us 9.173 us 9.173 us 9.173 us 9.173 us 9.173 us 9.173 us 9.173 us 1 1 torch.autograd.grad_mode.__init__ 9.052 us 9.052 us 9.052 us 9.052 us 9.052 us 9.052 us 9.052 us 9.052 us 1 1 torch.utils._contextlib.__new__ 4.910 us 4.910 us 4.910 us 4.910 us 4.910 us 4.910 us 4.910 us 4.910 us 1 1 Proxy 3.090 us 3.090 us 3.090 us 3.090 us 3.090 us 3.090 us 3.090 us 3.090 us 1 1 Net 1.953 us 1.953 us 1.953 us 1.953 us 1.953 us 1.953 us 1.953 us 1.953 us 1 1 dict.items 1.612 us 1.612 us 1.612 us 1.612 us 1.612 us 1.612 us 1.612 us 1.612 us 1 1 builtins.isinstance