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

Raise exception if doSampleBatch == True

Browse Source
This commit is contained in:
jens
2025-12-20 20:21:42 +01:00
parent 61c762150c
commit 7f4505c1f8
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/stack_factory.py
+6
View File
@@ -39,6 +39,12 @@ class StackFactory:
# Determine version by existence of keys # Determine version by existence of keys
params = layer["rbm"]["params"] params = layer["rbm"]["params"]
# signal that doSampleBatch will be ignored
if params["doSampleBatch"] == 1:
raise Exception(f"Error: Parameter \"doSampleBatch\" will be ignored. To continue, set \"doSampleBatch = False\"")
# import params
entity_params = EntityParams.from_dict(params) entity_params = EntityParams.from_dict(params)
training_params = TrainingParams.from_dict(params) training_params = TrainingParams.from_dict(params)