PyTorch-Mini-Tutorials
PyTorch-Mini-Tutorials copied to clipboard
GPU versions of example 3 to 6
Hi @vinhkhuc ,
Would you be interested in a PR for the cuda version of these 4 models? I made some simple modifications to make them work on GPUs, while trying to preserve the flow of the original non-GPU versions.
Example 1 & 2 probably doesn't need a GPU version since they train quite fast anyway.
Thx