PyTorch-Tutorial-2nd
PyTorch-Tutorial-2nd copied to clipboard
一些防呆设计,增加对苹果M系列芯片支持
感谢大佬分享,菜鸟最近正在研读
防呆设计
琢磨第8章的这段代码时发现几个默认设置很容易报错,略作修改,很小的改动,希望让本难以debug的同学们轻松一些 修改了--data-path为必须使用的参数,取消其默认设置,提示用户设置为自己的dataset路径 修改了--model默认参数为cpu,以免无gpu报错
增加对苹果M系列芯片支持
虽然只有一小撮人用MacBook学习pytorch,但也应该得到应有的支持😢
这个小改动很好,的确满足了不同设备的朋友,这样对mac设备更友好~