PyTorch(利用GPU加速训练)

上一篇文章只是5个数据,现在用100000数据进行训练;(代码如下)

PyTorch(利用GPU加速训练)_第1张图片

 效果:(GTX1060ti)

PyTorch(利用GPU加速训练)_第2张图片

cpu就不谈了。 

你可能感兴趣的:(Pytorch)