fix
This commit is contained in:
parent
09cbfad0ef
commit
c11b16dd1e
@ -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['loss'][1])
|
_run.info['history'] = str(history.history['loss'][-1])
|
||||||
|
|
||||||
|
|
||||||
@ex.main
|
@ex.main
|
||||||
|
Loading…
Reference in New Issue
Block a user