just loss
This commit is contained in:
parent
407d6187af
commit
09cbfad0ef
@ -46,7 +46,7 @@ def prepare_train_model(learning_rate, _run):
|
|||||||
|
|
||||||
model.save('grid-stability-dense.h5')
|
model.save('grid-stability-dense.h5')
|
||||||
|
|
||||||
_run['history'] = str(history.history[:, -1])
|
_run['history'] = str(history.history['loss'][1])
|
||||||
|
|
||||||
|
|
||||||
@ex.main
|
@ex.main
|
||||||
|
Loading…
Reference in New Issue
Block a user