double descent pytorch:在MNIST上使用PyTorch重现下降曲线 源码

双下降火炬 利用PyTorch在MNIST,特别是其上的实现/复制双下降曲线。 要重现图9(a)和(c),请运行以下命令。 通过标志--reuse控制是否使用权重复用。 在一个1080ti gpu上创建一个图形大约需要15个小时。 python mnist_double_descent.py --reuse --verbose 1 重用体重 没有重量重用