使用卷积神经网络进行图像分类

import paddle
import paddle.nn.functional as F
from paddle.vision.transforms import ToTensor
import numpy as np
import matplotlib.pyplot as plt

transform = ToTensor()
cifar10_train = paddle.vision.datasets.Cifar10(mode='train',
                                               transform=transform)
cifar10_test = paddle.vision.datasets.Cifar10(mode='test',
                                              transform=transform)

class MyNet(paddle.nn.Layer):
    def __init__(self, num_classes=1):
        super(MyNet, self).__init__()

        self.conv1 = paddle.nn.Conv2D(in_channels=3, out_channels=32, kernel_size=(2, 2))
        self.pool1 = paddle.nn.MaxPool2D(kernel_size=2, stride=2)
        #池化：降采样；此处采用最大值池化

        self.conv2 = paddle.nn.Conv2D(in_channels=32, out_channels=64, kernel_size=(2, 2))
        self