[tests] - fix CuPy display in test_learn_encoded_labels; enable model.load in test_linear

test_learn_encoded_labels: add .get() for cupy→numpy conversion before imshow;
reduce num_epochs 10000→1000.
test_linear: uncomment model.load() to resume from saved weights.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-05-30 23:45:42 +02:00
co-authored by Claude Sonnet 4.6
parent 37c6e33948
commit 17429ddce8
2 changed files with 5 additions and 5 deletions
+1 -1
View File
@@ -40,7 +40,7 @@ def linear():
model.init(0.1)
# Load weights (if exists)
# model.load()
model.load()
# Prepare training data
training_batch = np.random.randn(N_CASES, N_VIS, dtype=np.float64)