Advertisement

java深度学习之DJL搭建简单的CNN模型

阅读量:

1、算法描述

复制代码
   Model model = Model.newInstance(MODEL_NAME);

    
  
    
     // Block is a composable unit that forms a neural network; combine them like Lego blocks
    
     // to form a complex network
    
     Block block = new SequentialBlock()
    
             .add(Conv2d.builder()
    
                     //卷积核
    
                     .setKernelShape(new Shape(5, 5))
    
                     //步幅
    
                     .optStride(new Shape(5, 5))
    
                     //padding
    
                     .optPadding(new Shape(2, 2))
    
                     //过滤器
    

全部评论 (0)

还没有任何评论哟~