自己的一个熟悉API小模块,主要使用Pytorch搭建了ResNet50网络,受限于数据集的大小限制,训练后的模型精度只有60%,模型在分辨“女包”和“奢侈品”上表现很差。熟悉Pytorch的使用,网络搭建,数据集预处理,模型参数调优。
评论