This commit is contained in:
jakubknczny 2021-05-16 16:07:47 +02:00
parent 09cbfad0ef
commit c11b16dd1e

View File

@ -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