Skip to content

Commit 8b2674b

Browse files
Migrate to Keras 1.2x
1 parent 75d66a9 commit 8b2674b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

plasma/models/mpi_runner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -479,10 +479,10 @@ def mpi_train(conf,shot_list_train,shot_list_validate,loader, callbacks_list=Non
479479

480480
callbacks = mpi_model.build_callbacks(conf,callbacks_list)
481481

482-
callbacks._set_model(mpi_model.model)
482+
callbacks.set_model(mpi_model.model)
483483
callback_metrics = conf['callbacks']['metrics']
484484

485-
callbacks._set_params({
485+
callbacks.set_params({
486486
'nb_epoch': num_epochs,
487487
'metrics': callback_metrics,
488488
})

0 commit comments

Comments
 (0)