modified conv2d.yaml