pytorch灰度图训练