better error message

simon-larsson · simon-larsson · commit d4d27c3c2eaa · 2021-05-02T19:58:57.000+02:00
diff --git a/swa/callback.py b/swa/callback.py
@@ -99,7 +99,7 @@ def on_train_begin(self, logs=None):
 
             if self.has_batch_norm and self.batch_size is None:
                 raise ValueError(
-                    '"batch_size" needs to be set for models with batch normalization.'
+                    '"batch_size" needs to be set for models with batch normalization layers.'
                 )
 
         def on_epoch_begin(self, epoch, logs=None):

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ def on_train_begin(self, logs=None):`
`99`	`99`
`100`	`100`	`if self.has_batch_norm and self.batch_size is None:`
`101`	`101`	`raise ValueError(`
`102`		`- '"batch_size" needs to be set for models with batch normalization.'`
	`102`	`+ '"batch_size" needs to be set for models with batch normalization layers.'`
`103`	`103`	`)`
`104`	`104`
`105`	`105`	`def on_epoch_begin(self, epoch, logs=None):`