深度学习-程序过早停止诊断

最新推荐文章于 2025-04-04 03:56:49 发布

原创

最新推荐文章于 2025-04-04 03:56:49 发布 · 2.9k 阅读

0 ·

CC 4.0 BY-SA版权

本文主要探讨深度学习实战中遇到的程序过早停止问题，虽然内容基础，但在实际应用中具有重要意义。文章将对比以往知识，对关键区别进行注解。

内容不难，但是在实战中很有用，大部分内容和之前一样，特别的地方会做注解

public class EarlyStoppingMNIST {

    public static void main(String[] args) throws Exception {

        //Configure network:
        int nChannels = 1;
        int outputNum = 10;
        int batchSize = 25;
        int iterations = 1;
        int seed = 123;
        MultiLayerConfiguration.Builder builder = new NeuralNetConfiguration.Builder()
                .seed(seed)
                .iterations(iterations)
                .regularization(true).l2(0.0005)
                .learningRate(0.02)
                .weightInit(WeightInit.XAVIER)
                .activation("relu")
                .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
                .updater(Updater.NESTEROVS).momentum(0.9)
                .list()
                .l