[CheckpointStatus] - add checkpoint status handler; Model.train() accepts status param

status.py: add CheckpointStatus(save_fn, update_interval) — prints progress
and calls save_fn at every report interval to persist model state mid-training.
model.py: Model.train() now accepts an optional Status instance; defaults to
plain Status() if none provided.
test_faces_sub_image.py: use CheckpointStatus(model.save) during training.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-05-30 22:42:01 +02:00
co-authored by Claude Sonnet 4.6
parent 2f9fe1b66b
commit 37c6e33948
3 changed files with 17 additions and 3 deletions
+2 -1
View File
@@ -7,6 +7,7 @@ from rbm.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.image import SubImage, normalize
from rbm.matrix import Mat, np, convert
from rbm.status import CheckpointStatus
DATA_DIR = '/media/jens/cifs/bilder/MachineVision/Caltech_WebFaces/'
PATCH = 32
@@ -214,7 +215,7 @@ if __name__ == '__main__':
print(f'Loading {args.n_images} training images (stride={STRIDE})...')
train_patches = load_patches(DATA_DIR, args.n_images)
print(f'Training on {train_patches.shape[0]} patches ({N_VIS}-dim each)')
model.train(train_patches)
model.train(train_patches, status=CheckpointStatus(save_fn=model.save))
model.save()
# Show learned weight filters