Skip to content

Commit 10d02b3

Browse files
committed
fix
1 parent 649c970 commit 10d02b3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

model_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def __init__(self, model, root_name, ids,
101101
# can changed during runtime
102102
write_graph = False, # TensorBoard params
103103
write_images = False, #
104-
logs_root = 'None', # TensorBoard logs
104+
logs_root = None, # TensorBoard logs
105105
models_root = 'models/', # saved models path
106106
features_root = 'features/', # saved features path (by `save_activations`)
107107
gen_class = None, # generator class inferred from self.gen_params.data_path
@@ -285,7 +285,7 @@ def train(self, train_gen=True, valid_gen=True,
285285
if lr: self.params.lr = lr
286286
self.params.optimizer =\
287287
update_config(self.params.optimizer,
288-
lr=self.params.lr)
288+
lr=self.params.lr)
289289
self.compile()
290290

291291
if self.verbose:

0 commit comments

Comments
 (0)