blob: 5478903bf0ea355b1f2b47e9e914739e91858cd3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
Testing optimized SpikingVGG forward...
Testing on device: cuda
Using: Global delta + Global renorm (Option 1 - textbook LE)
Model depth: 6 conv layers
Parameters: 1,187,274
[Test 1] Forward without Lyapunov...
Logits shape: torch.Size([8, 10]) ✓
[Test 2] Forward with Lyapunov...
Logits shape: torch.Size([8, 10]) ✓
Lyapunov exponent: 3.3281 ✓
[Test 3] Backward pass...
Loss: 5.3691 ✓
Gradient norm: 25.8283 ✓
No NaN gradients ✓
[Test 4] Multiple training steps...
Step 1: loss=7.0680, λ=3.0538
Step 2: loss=7.3570, λ=3.2788
Step 3: loss=8.4584, λ=2.8295
Step 4: loss=5.7036, λ=3.3123
Step 5: loss=7.4161, λ=3.4701
==================================================
ALL TESTS PASSED!
==================================================
Done!
|
