switch pytorch code

切换到pytorch代码块,而不是默认的块,仅在第一次进入界面时生效

这些是此脚本的所有版本。 只显示代码变更过的版本。