Neural-ODE

Neural ODE pytorch实现MNIST数据集图像分类
NeuralODE_train_1.py是在MNIST数据集上从头开始训练（MNIST数据集会自动下载）
NeuralODE_train_2.py实现的功能是可以在已有模型（epoch0050_neural_ode_mnist.pth）基础上继续训练
NeuralODE_test.py是测试代码

注意：

torch安装推荐：pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu118

测试结果：