PyTorch-Mini-Tutorials icon indicating copy to clipboard operation
PyTorch-Mini-Tutorials copied to clipboard

GPU versions of example 3 to 6

Open esvhd opened this issue 8 years ago • 0 comments

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

esvhd avatar Jul 20 '17 21:07 esvhd