Minor fixes

This commit is contained in:
Francois Chollet 2023-06-29 12:57:32 -07:00
parent ba1442a22f
commit 7f37477770
2 changed files with 1 additions and 4 deletions

@ -10,9 +10,6 @@ from keras_core import ops
from keras_core import optimizers
keras_core.config.disable_traceback_filtering()
class MyDense(layers.Layer):
def __init__(self, units, name=None):
super().__init__(name=name)

@ -14,7 +14,7 @@ def torch_seed_generator(seed):
seed_val, _ = draw_seed(seed)
device = get_device()
if device == "meta":
# Generator is not support by the meta device.
# Generator is not supported by the meta device.
return None
generator = torch.Generator(device=get_device())
generator.manual_seed(int(seed_val))