PyTorch-Mini-Tutorials
PyTorch-Mini-Tutorials copied to clipboard
Minimal tutorials for PyTorch
doing .forward I believe is not recommended because it doesn't end up using masks or handle or something like that I forget...just heads up
https://github.com/vinhkhuc/PyTorch-Mini-Tutorials/blob/cbf48ed33aa32fb58bfee594e4b7a169bb95e283/2_logistic_regression.py#L20
https://github.com/vinhkhuc/PyTorch-Mini-Tutorials/blob/cbf48ed33aa32fb58bfee594e4b7a169bb95e283/2_logistic_regression.py#L47
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...