摘要: 使用VGG模型进行猫狗大战 首先判断是否存在GPU设备 2. 数据处理 datasets 是 torchvision 中的一个包,可以用做加载图像数据。它可以以多线程(multi-thread)的形式从硬盘中读取数据,使用 mini-batch 的形式,在网络训练中向 GPU 输送。在使用CNN处理 阅读全文
posted @ 2021-10-24 19:44 Drew。 阅读(36) 评论(0) 推荐(0)