small fix for the autoclustering_xla.ipynb

This commit is contained in:
Yong Wu 2020-08-29 00:56:22 +08:00 committed by GitHub
parent ccd2fff028
commit 64138d50ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -169,7 +169,7 @@
" model.set_weights(initial_weights)\n",
"\n",
"warmup(model, x_train, y_train, x_test, y_test)\n",
"%time train_model(model, x_train, y_train, x_test, y_test)\n",
"train_model(model, x_train, y_train, x_test, y_test)\n",
"\n",
"scores = model.evaluate(x_test, y_test, verbose=1)\n",
"print('Test loss:', scores[0])\n",