MobileNet 实现对CIFAR-10数据集训练(pytorch)
importtorchimporttorchvisionfromtorchvisionimportdatasets,transformsfromtorch.utils.dataimportDataLoaderfromtorchimportnnimporttorch.nn.functionalasFimportnumpyasnpfromtorch.utils.dataimportDataset,DataLoader,TensorData...
原创
2020-08-26 10:15:01 ·
1402 阅读 ·
0 评论