Cannot measure CPU usage on MNIST example

7

@ptrblck
I have one question and one request. Would you check it?

  1. Question about Memory Usage for torch.utils.bottleneck with MNIST.
    Training MNIST with --epochs 3 --no-cuda works fine, but failed with --epochs 4 --no-cuda on my server with 16GB-RAM machine. It seems like RAM memory (RES) consumption problem. I guess it needs 64GB-RAM for MNIST Training sample with torch.utils.bottleneck.
    Is my guess correct?

  2. As for the document on torch.utils.bottleneck
    It is helpful to write a document for trace customization (topk etc).

cp torch/utils/bottleneck/__main__.py bottleneck.py
pytorch bottleneck.py /script/path/to/source/script.py [args]