small fix for the autoclustering_xla.ipynb
This commit is contained in:
parent
ccd2fff028
commit
64138d50ce
@ -169,7 +169,7 @@
|
|||||||
" model.set_weights(initial_weights)\n",
|
" model.set_weights(initial_weights)\n",
|
||||||
"\n",
|
"\n",
|
||||||
"warmup(model, x_train, y_train, x_test, y_test)\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",
|
"\n",
|
||||||
"scores = model.evaluate(x_test, y_test, verbose=1)\n",
|
"scores = model.evaluate(x_test, y_test, verbose=1)\n",
|
||||||
"print('Test loss:', scores[0])\n",
|
"print('Test loss:', scores[0])\n",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user