Skip to content

Commit

Permalink
Update alexnet.py
Browse files Browse the repository at this point in the history
You may want tensorboard_verbose=2 initially to see whats going on with your model, but you defintiely don't want to leave it there for the whole training, it significantly impacts training.
  • Loading branch information
Sentdex authored May 9, 2017
1 parent 2535835 commit 19468f9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion alexnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def alexnet(width, height, lr):
learning_rate=lr, name='targets')

model = tflearn.DNN(network, checkpoint_path='model_alexnet',
max_checkpoints=1, tensorboard_verbose=2, tensorboard_dir='log')
max_checkpoints=1, tensorboard_verbose=0, tensorboard_dir='log')

return model

Expand Down

0 comments on commit 19468f9

Please sign in to comment.