Minor fixes
This commit is contained in:
parent
ba1442a22f
commit
7f37477770
@ -10,9 +10,6 @@ from keras_core import ops
|
|||||||
from keras_core import optimizers
|
from keras_core import optimizers
|
||||||
|
|
||||||
|
|
||||||
keras_core.config.disable_traceback_filtering()
|
|
||||||
|
|
||||||
|
|
||||||
class MyDense(layers.Layer):
|
class MyDense(layers.Layer):
|
||||||
def __init__(self, units, name=None):
|
def __init__(self, units, name=None):
|
||||||
super().__init__(name=name)
|
super().__init__(name=name)
|
||||||
|
@ -14,7 +14,7 @@ def torch_seed_generator(seed):
|
|||||||
seed_val, _ = draw_seed(seed)
|
seed_val, _ = draw_seed(seed)
|
||||||
device = get_device()
|
device = get_device()
|
||||||
if device == "meta":
|
if device == "meta":
|
||||||
# Generator is not support by the meta device.
|
# Generator is not supported by the meta device.
|
||||||
return None
|
return None
|
||||||
generator = torch.Generator(device=get_device())
|
generator = torch.Generator(device=get_device())
|
||||||
generator.manual_seed(int(seed_val))
|
generator.manual_seed(int(seed_val))
|
||||||
|
Loading…
Reference in New Issue
Block a user