summaryrefslogtreecommitdiff
path: root/training/train_cnn.py
AgeCommit message (Expand)Author
24 hoursfix: Move sigmoid activation to call site in CNN layer shaderskal
24 hoursfix: Replace clamp with sigmoid in CNN final layerskal
24 hoursfeat: Add early stopping to CNN trainingskal
25 hoursfix: CNN training/inference to match WGSL sliding windowskal
25 hoursformat .wgsl layer code (cosmetics)skal
33 hoursfix: Use patch-based inference to match CNN training distributionskal
34 hoursopt: Move invariant in1 calculation outside CNN convolution loopsskal
35 hoursopt: Vec4-optimize CNN convolution shaders for SIMDskal
35 hoursfeat: Add salient-point patch extraction for CNN trainingskal
37 hoursfix: Correct UV coordinate computation to match PyTorch linspaceskal
37 hoursfix: Add clamp to CNN final layer to match PyTorch trainingskal
37 hoursrefactor: Optimize CNN grayscale computationskal
37 hoursupdate train_cnn.py and shaderskal
38 hoursfeat: Add inference mode to train_cnn.py for ground truth generationskal
38 hoursfix: CNN training normalization pipeline consistencyskal
39 hoursrefactor: Optimize CNN normalization to eliminate redundant conversionsskal
40 hoursfix: Support variable kernel sizes in CNN layer generationskal
41 hoursfeat: CNN RGBD→grayscale with 7-channel augmented inputskal
45 hoursfeat: Add multi-layer CNN support with framebuffer capture and blend controlskal
47 hoursfeat: Add checkpointing support to CNN training scriptskal
47 hoursfix: Auto-expand single kernel size to all layers in training scriptskal
48 hoursfeat: Add coordinate-aware CNN layer 0 for position-dependent stylizationskal