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)
还没有任何评论哟~
