今天我花了一天时间把MAIR从caffe转移到了pytorch。相对于caffe，pytorch的代码确实好写很多啊！

pytorch的数据集处理部分有一个暗坑，即label应该是类别对应的数字，比如["12", "15", "17"]这三个标签，必须转成[0, 1, 2]，这个甚至是在cuda代码里限制住的。

另外，torchvision的代码最好还是都过一遍，都很值得参考呀。

小于今天胃疼。也差不多定下来了老师，希望姜老师是个人。