summaryrefslogtreecommitdiff
path: root/training/train_cnn.py
AgeCommit message (Expand)Author
34 hoursdocs: Update CNN comments and add bias fix summaryskal
35 hoursfix: CNN bias accumulation and output format improvementsskal
41 hoursrefactor: Use linspace(-1,1) directly for coordsskal
41 hoursfix: Compute gray from [0,1] RGB in CNN shader generatorskal
47 hoursadd --save-intermediates to train.py and cnn_testskal
48 hoursfix: Move sigmoid activation to call site in CNN layer shaderskal
48 hoursfix: Replace clamp with sigmoid in CNN final layerskal
2 daysfeat: Add early stopping to CNN trainingskal
2 daysfix: CNN training/inference to match WGSL sliding windowskal
2 daysformat .wgsl layer code (cosmetics)skal
2 daysfix: Use patch-based inference to match CNN training distributionskal
2 daysopt: Move invariant in1 calculation outside CNN convolution loopsskal
2 daysopt: Vec4-optimize CNN convolution shaders for SIMDskal
2 daysfeat: Add salient-point patch extraction for CNN trainingskal
3 daysfix: Correct UV coordinate computation to match PyTorch linspaceskal
3 daysfix: Add clamp to CNN final layer to match PyTorch trainingskal
3 daysrefactor: Optimize CNN grayscale computationskal
3 daysupdate train_cnn.py and shaderskal
3 daysfeat: Add inference mode to train_cnn.py for ground truth generationskal
3 daysfix: CNN training normalization pipeline consistencyskal
3 daysrefactor: Optimize CNN normalization to eliminate redundant conversionsskal
3 daysfix: Support variable kernel sizes in CNN layer generationskal
3 daysfeat: CNN RGBD→grayscale with 7-channel augmented inputskal
3 daysfeat: Add multi-layer CNN support with framebuffer capture and blend controlskal
3 daysfeat: Add checkpointing support to CNN training scriptskal
3 daysfix: Auto-expand single kernel size to all layers in training scriptskal
3 daysfeat: Add coordinate-aware CNN layer 0 for position-dependent stylizationskal