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
|
||||
|
||||
|
||||
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))
|
||||
|
Loading…
Reference in New Issue
Block a user