examples

examples copied to clipboard

Your issue may already be reported! Please search on the [issue tracker](https://github.com/pytorch/examples/issues) before creating one. ## Context * Pytorch version: 2.7.1 * Operating System and version: Ubuntu ## Your Environment...

zhc7

Use of torch.inference_mode() instead of torch.no_grad() for inference

3

Replaced the `torch.no_grad()` context manager with the `torch.inference_mode()` context manager in the test function. Why? `torch.inference_mode()` is more efficient as it reduces memory usage, inference speed and it disabled autograd...

thomastschinkel

mnist: support --model-path and --load-model

4

If specify --load-model will only run test(), if specify --model-path=A, the saved model will be renamed to A. Example: python main.py --save-model --model-path=mnist_cnn.pt python main.py --load-model=mnist_cnn.pt

Rtoax

mnist_forward_forward: Load dataset from comment path

2

Load the same dataset file as other MNIST test cases.

Rtoax

Your issue may already be reported! Please search on the [issue tracker](https://github.com/pytorch/examples/issues) before creating one. ## Context * Pytorch version: 2.6.0 * Operating System and version: linux ## Your Environment...

Nju-Ben

Hello, I use the command "python main.py -a resnet50 --dist-url 'tcp://127.0.0.1:60000/' --dist-backend 'nccl' --multiprocessing-distributed --world-size 1 --rank 0 /my_data_dir/" train and test resnet50 on a single node with 8 GPUs....

sdwhzh