MobileNet 实现对CIFAR-10数据集训练(pytorch)
importtorch
importtorchvision
fromtorchvisionimportdatasets,transforms
fromtorch.utils.dataimportDataLoader
fromtorchimportnn
importtorch.nn.functionalasF
importnumpyasnp
fromtorch.utils.dataimportDataset,DataLoader,TensorData...
原创
2020-08-26 10:15:01 ·
1559 阅读 ·
0 评论