examples
examples copied to clipboard
pytorch
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
I am running test under : https://github.com/pytorch/examples/tree/main/imagenet Does the main.py support FP16(amp)?like arguments： --amp Thx
I'm trying to use this code as a starting point for building GANs from my own image data-- 512x512 grayscale images. If I change any of the default arguments (e.g....
Currently, ``` total ```(https://github.com/pytorch/examples/blob/main/imagenet/main.py#L419) is at cpu but it must be CUDA to compute ``` dist.all_reduce() ```. Otherwise it will raise a ``` RuntimeError: Tensors must be CUDA and dense...
Currently, ```total```(https://github.com/pytorch/examples/blob/main/imagenet/main.py#L419) is at cpu but it must be CUDA to compute ```dist.all_reduce()```. Otherwise it will raise a ```RuntimeError: Tensors must be CUDA and dense```
Your issue may already be reported! Please search on the [issue tracker](https://github.com/pytorch/serve/examples) before creating one. ## Context th.multiprocessing seems to not send tensor data to spawned processes on my setup....
Following the discussions in #995 and #862, This PR adds a small usage example into the ImageNet README. One note, however, is that specifying ```--gpu``` will trigger the print in...
This example shows training a HF T5 model with FSDP to be used with its [tutorial](https://github.com/pytorch/tutorials/pull/1959#discussion_r913870745)
### Description Let's use modern style for modern code. These changes remove python 2 syntax all over the place in favor of modern python 3 syntax first released in 2008....
