转战pytorch（2）——“简单”训练