File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -189,8 +189,7 @@ def __init__(
189189 }
190190
191191 def get_head_shapes_from_target_shape (self , target_shape : Shape ):
192- # keras.saving.load_model sometimes passes target_shape as a list.
193- # This is why I force a conversion to tuple here.
192+ # keras.saving.load_model sometimes passes target_shape as a list, so we force a conversion
194193 target_shape = tuple (target_shape )
195194 return dict (value = target_shape [1 :])
196195
@@ -255,8 +254,7 @@ def get_config(self):
255254 return base_config | self .config
256255
257256 def get_head_shapes_from_target_shape (self , target_shape : Shape ):
258- # keras.saving.load_model sometimes passes target_shape as a list.
259- # This is why I force a conversion to tuple here.
257+ # keras.saving.load_model sometimes passes target_shape as a list, so we force a conversion
260258 target_shape = tuple (target_shape )
261259 return dict (value = (len (self .q ),) + target_shape [1 :])
262260
You can’t perform that action at this time.
0 commit comments