diff --git a/src/rbm/stack_factory.py b/src/rbm/stack_factory.py index b56c8a5..c20e09b 100644 --- a/src/rbm/stack_factory.py +++ b/src/rbm/stack_factory.py @@ -39,6 +39,12 @@ class StackFactory: # Determine version by existence of keys 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) training_params = TrainingParams.from_dict(params)