diff options
Diffstat (limited to 'workspaces/main/shaders/cnn/cnn_layer.wgsl')
| -rw-r--r-- | workspaces/main/shaders/cnn/cnn_layer.wgsl | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/workspaces/main/shaders/cnn/cnn_layer.wgsl b/workspaces/main/shaders/cnn/cnn_layer.wgsl index d33a301..48bdcc6 100644 --- a/workspaces/main/shaders/cnn/cnn_layer.wgsl +++ b/workspaces/main/shaders/cnn/cnn_layer.wgsl @@ -8,7 +8,6 @@ #include "common_uniforms" #include "cnn_activation" #include "cnn_conv3x3" -#include "cnn_conv5x5" #include "cnn_weights_generated" struct CNNLayerParams { @@ -43,7 +42,7 @@ struct CNNLayerParams { result = cnn_tanh(result); } else if (params.layer_index == 1) { - result = cnn_conv5x5_7to4(txt, smplr, uv, uniforms.resolution, + result = cnn_conv3x3_7to4(txt, smplr, uv, uniforms.resolution, gray, weights_layer1); result = cnn_tanh(result); // Keep in [-1,1] } |