Francois Chollet
468523a43e
Fix code format
2023-06-27 16:23:31 -07:00
Chen Qian
eee3ca3c9a
Fix mixed precision issue in JAX backend ( #410 )
...
* init
* Add logging
* minor
* port kerasnlp
* fix jax error
* training fix
* wrap up
* rebase main
* fix mixed precision
* fix comments
* fix test failures
* fix type casting
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-scratch.us-west1-a.c.keras-team-gcp.internal>
2023-06-27 16:22:52 -07:00
Francois Chollet
ae3102106f
Merge branch 'main' of github.com:keras-team/keras-core
2023-06-26 13:46:17 -07:00
Francois Chollet
4400b55fd8
Add masking guide, callbacks guide
2023-06-26 13:46:10 -07:00
Chen Qian
0716889b9e
Update command of efficient_net_image_classification_benchmark.py
2023-06-26 13:43:45 -07:00
Francois Chollet
36a4a83220
Minor fixes to model summaries
2023-06-24 14:51:19 -07:00
Chen Qian
dcf5e7b209
Add BERT benchmark on GLUE/MRPC ( #381 )
...
* init
* Add logging
* minor
* port kerasnlp
* fix jax error
* training fix
* wrap up
* rebase main
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-scratch.us-west1-a.c.keras-team-gcp.internal>
2023-06-23 23:14:16 -07:00
Chen Qian
8a172286dd
Add EfficientNet benchmark ( #394 )
...
* init
* refactor
* minor improvements
* Remove softmax activation
* small
* rename
* redo the tf change
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-scratch.us-west1-a.c.keras-team-gcp.internal>
2023-06-23 15:56:10 -07:00
Haifeng Jin
9de576d905
model.call() func compiled ( #331 )
2023-06-22 12:02:22 -07:00
Francois Chollet
79a6464978
Merge branch 'main' of github.com:keras-team/keras-core
2023-06-12 16:37:12 -07:00
Francois Chollet
06225c04cd
Fix Aurelien's new issue
2023-06-12 16:37:00 -07:00
Chen Qian
39bbd2a03d
Refactor layer benchmark ( #333 )
...
* some benchmark setup
* fix something
* fix rnn benchmark
* Fix format
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-4.us-west1-a.c.keras-team-gcp.internal>
2023-06-12 16:02:14 -07:00
Francois Chollet
1239ec8b04
Merge branch 'main' of github.com:keras-team/keras-core
2023-06-08 19:16:24 -07:00
Francois Chollet
fabbf81880
Remove unused files
2023-06-08 19:16:18 -07:00
Chen Qian
61b76f1655
Add benchmark for all leftover layers ( #308 )
...
* add more benchmark
* Add benchmark coverage for all layers
2023-06-08 19:05:20 -07:00
Chen Qian
2a6d247f78
Add torch benchmark to compare performance when using torch CTL ( #287 )
...
* Eager torch benchmark
* small fix
* add a simpler benchmark
* small fix
* refactor the code
* refactor
* Add readme
2023-06-07 18:38:56 -07:00
Chen Qian
289dc028a9
Optimize the way to convert array data into Tf dataset during training ( #267 )
...
* initial
* fix format
* skip the broken test in TF
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-4.us-west1-a.c.keras-team-gcp.internal>
2023-06-05 18:41:35 -07:00
Chen Qian
35082455e5
Add more layer benchmarks ( #245 )
...
* initials
* add benchmark class
* add train benchmark
* Add conv benchmark
* flag
* fix comments
* docstring
* remove redundant flags
* More benchmarks
* better
* add more benchmarks
* remove weird files
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu3.us-west1-a.c.keras-team-gcp.internal>
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-4.us-west1-a.c.keras-team-gcp.internal>
2023-06-05 11:05:12 -07:00
Chen Qian
ebe9ee5a1a
Add benchmark for layers ( #231 )
...
* initials
* add benchmark class
* add train benchmark
* Add conv benchmark
* flag
* fix comments
* docstring
* remove redundant flags
---------
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu3.us-west1-a.c.keras-team-gcp.internal>
Co-authored-by: chenmoneygithub <chenmoney@chenmoney-gpu-4.us-west1-a.c.keras-team-gcp.internal>
2023-05-31 21:46:39 -07:00