SPHIRE Workflow
Sidebar
pipeline:window:cryolo:training
This version (2020/03/16 15:21) is a draft.
Approvals: 0/1The Previously approved version (2019/09/18 08:36) is available.
Approvals: 0/1The Previously approved version (2019/09/18 08:36) is available.
Now you are ready to train the model. In case you have multiple GPUs, you should first select a free GPU. The following command will show the status of all GPUs:
nvidia-smi
For this tutorial, we assume that you have either a single GPU or want to use GPU 0.
Use a different or multiple GPUs
In the “Optional arguments” tab you can change the GPU that should be used by crYOLO. If you have multiple GPUs (e.g. nvidia-smi lists GPU 0 and GPU 1) you can also use both by setting the GPU argument to '0 1'.
In the GUI you have to fill in the mandatory fields:
The default number of warmup epochs1) is fine as long as you don't want to refine an existing model. During the warmup training epochs it will not try to estimate the size of your particle, which helps crYOLO to converge.
When does crYOLO stop the training?
When you start the training, it will stop when the “loss” metric on the validation data does not improve 10 times in a row. This is typically enough. In case you want to give the training more time to find the best model can increase the “not changed in a row” parameter to a higher value by setting the early argument in the “Optional arguments” to, for example, 15.
The final model will be written to disk as specified in saved_weights_name in your configuration file.
► Now press the [Start] button to start the training.
Alternative: Train crYOLO using the command line
1)
One epoch is a complete pass through the training data.
pipeline/window/cryolo/training.txt · Last modified: 2020/03/16 15:21 by twagner
Page Tools
Except where otherwise noted, content on this wiki is licensed under the following license: GNU Free Documentation License 1.3
