jensandClaude Sonnet 4.6 2f9fe1b66b [bugfix] - support large CPU-side datasets; fix numpy/CuPy interop
train.py: add _to_gpu() helper; convert mini-batches from CPU numpy to GPU
CuPy on the fly so large datasets can stay in RAM. Use numpy permutation for
numpy batches during shuffle. Limit status err_rms to first 5000 samples to
avoid GPU OOM on status checks.

model.py: replace np.copy() with np.array() — cupy.copy() does not convert
numpy arrays to CuPy; cupy.array() does. Fixes entity.forward() crash after
training when batch was passed as numpy.

test_faces_sub_image.py: load_patches normalizes on CPU and returns numpy,
avoiding multi-copy GPU OOM for large datasets (500 images ≈ 1.4 GB).
show_reconstructions converts numpy patches to CuPy before inference.
Add --n_images CLI arg.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-30 22:26:06 +02:00
2025-12-19 11:11:00 +01:00
2026-01-11 11:23:35 +01:00
2026-01-10 21:44:03 +01:00
2025-12-19 15:20:04 +01:00
2026-01-11 11:23:35 +01:00
2026-01-10 19:58:39 +01:00
2025-12-18 07:56:45 +01:00
2026-01-06 18:00:26 +01:00
2025-12-16 21:46:00 +01:00
S
Description
Restricted Boltzmann Machine (RBM) for python
Readme
65 MiB
Languages
Jupyter Notebook 96.9%
Python 3.1%