-
Notifications
You must be signed in to change notification settings - Fork 64
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Tensorboard #138
base: master
Are you sure you want to change the base?
Tensorboard #138
Conversation
Tensorboard generates a ridiculous amount of event logs... more than 1.2GB of logs per epoch only for a basic transformer model. For instance after 4 epochs:
Given that effect (like with a 10-fold cross-validation training), the tensorflow callbacks need to be defined at the application level, so that it is a user choice and not systematically used by the core library. |
delft/sequenceLabelling/models.py
Outdated
@@ -870,10 +870,12 @@ def __init__(self, config, ntags=None, load_pretrained_weights=True, local_path: | |||
x = Concatenate()([text_embedding_layer, features_embedding_out]) | |||
x = Dropout(config.dropout)(x) | |||
|
|||
''' |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Was this intentional?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nope I was testing features with architecture having a transformer layer, because it's not working as expected.
I am still working on this branch, I should have put a draft flag !
Looks good. Indeed, this tool can fill up the disk space faster than docker... 🎉 |
FYI 1 run of 10-fold crossvalidation weight ... 82GB 😂
|
This PR setup DeLFT to use tensorboard: logs, callbacks...
The idea is to cover the interesting metrics so that everything is visualized on Tensorboard.
How to use: basically nothing special, start a training and launch:
> tensorboard --logdir logs/summaries
(tensorboard is already installed)
Then open http://localhost:6006/
TODO: