lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:keras-team/keras-core

Browse Source
This commit is contained in:
Francois Chollet 2023-06-26 13:46:17 -07:00
parent 4400b55fd8
commit ae3102106f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/model_benchmark/efficient_net_image_classification_benchmark.py
View File

@ -3,7 +3,7 @@
To run the benchmark, make sure you are in model_benchmark/ directory, and run
the command below:
python3 -m model_benchmark.resnet_image_classification_benchmark \
python3 -m model_benchmark.efficient_net_image_classification_benchmark \
--epochs=2 \
--batch_size=32
"""

2
keras_core/backend/torch/numpy.py
View File

@ -661,7 +661,7 @@ def repeat(x, repeats, axis=None):
x = KerasTensor(x.shape, standardize_dtype(x.dtype))
outputs = repeat(x, repeats, axis=axis)
print(outputs)
return torch.empty(
size=outputs.shape,
dtype=to_torch_dtype(outputs.dtype),