PyTorch-Tutorial-2nd icon indicating copy to clipboard operation
PyTorch-Tutorial-2nd copied to clipboard

一些防呆设计,增加对苹果M系列芯片支持

Open whitewatercn opened this issue 1 year ago • 1 comments

感谢大佬分享,菜鸟最近正在研读

防呆设计

琢磨第8章的这段代码时发现几个默认设置很容易报错,略作修改,很小的改动,希望让本难以debug的同学们轻松一些 修改了--data-path为必须使用的参数,取消其默认设置,提示用户设置为自己的dataset路径 修改了--model默认参数为cpu,以免无gpu报错

增加对苹果M系列芯片支持

虽然只有一小撮人用MacBook学习pytorch,但也应该得到应有的支持😢

whitewatercn avatar Jun 10 '24 12:06 whitewatercn

这个小改动很好,的确满足了不同设备的朋友,这样对mac设备更友好~

TingsongYu avatar Jun 16 '24 10:06 TingsongYu