diff options
Diffstat (limited to 'workspaces/main/shaders/cnn/cnn_conv3x3.wgsl')
| -rw-r--r-- | workspaces/main/shaders/cnn/cnn_conv3x3.wgsl | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/workspaces/main/shaders/cnn/cnn_conv3x3.wgsl b/workspaces/main/shaders/cnn/cnn_conv3x3.wgsl index 79b0350..00eae22 100644 --- a/workspaces/main/shaders/cnn/cnn_conv3x3.wgsl +++ b/workspaces/main/shaders/cnn/cnn_conv3x3.wgsl @@ -138,5 +138,5 @@ fn cnn_conv3x3_7to1( } } - return sum; // Output in [-1,1] + return clamp(sum, 0.0, 1.0); // Match PyTorch clamp } |