jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[faces_sub_image] - add --num_epochs CLI arg

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
jens
2026-05-30 20:38:33 +02:00
co-authored by Claude Sonnet 4.6
parent edeeb1773b
commit 96f271f160
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tests/test_faces_sub_image.py
+5 -3
View File
@@ -19,12 +19,12 @@ N_IMAGES = 50
class TestModel(Model):
def __init__(self, name: str, work_dir: str = '.', l1_lambda: float = 0.0):
def __init__(self, name: str, work_dir: str = '.', l1_lambda: float = 0.0, num_epochs: int = 1000):
super().__init__(name, work_dir)
self.unit1 = Entity(
(N_VIS, N_HID),
EntityParams(do_gaussian_visible=True, do_gaussian_hidden=False),
TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000, mini_batch_size=1000,
TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=num_epochs, mini_batch_size=1000,
l1_lambda=l1_lambda)
)
@@ -178,6 +178,8 @@ if __name__ == '__main__':
help='Use single-channel grayscale patches (default: false)')
ap.add_argument('--l1_lambda', type=float, default=0.0,
help='L1 regularisation strength (default: 0.0)')
ap.add_argument('--num_epochs', type=int, default=1000,
help='Number of training epochs (default: 1000)')
args = ap.parse_args()
if args.grayscale:
@@ -188,7 +190,7 @@ if __name__ == '__main__':
prj_name = 'faces_sub_image_gray' if args.grayscale else 'faces_sub_image'
work_dir = 'results'
model = TestModel(prj_name, work_dir, l1_lambda=args.l1_lambda)
model = TestModel(prj_name, work_dir, l1_lambda=args.l1_lambda, num_epochs=args.num_epochs)
model.init(0.001)
if args.load_model: