Minor fixes
This commit is contained in:
parent
b14eaaf5da
commit
a33eb25ca9
@ -46,7 +46,7 @@ elif backend.backend() == "torch":
|
|||||||
from keras_core.backend.torch.layer import TorchLayer as BackendLayer
|
from keras_core.backend.torch.layer import TorchLayer as BackendLayer
|
||||||
else:
|
else:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
f"Backend '{backend.backend()}' must implement the LayerMixin class."
|
f"Backend '{backend.backend()}' must implement a layer mixin class."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -5,8 +5,8 @@ from keras_core.layers.pooling.base_global_pooling import BaseGlobalPooling
|
|||||||
|
|
||||||
@keras_core_export(
|
@keras_core_export(
|
||||||
[
|
[
|
||||||
"keras_core.layers.GlobalMaxPooling1D",
|
"keras_core.layers.GlobalMaxPooling2D",
|
||||||
"keras_core.layers.GlobalMaxPool1D",
|
"keras_core.layers.GlobalMaxPool2D",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
class GlobalMaxPooling2D(BaseGlobalPooling):
|
class GlobalMaxPooling2D(BaseGlobalPooling):
|
||||||
|
Loading…
Reference in New Issue
Block a user