This is an old revision of the document!
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.
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.
The final model will be written to disk as specified in saved_weights_name in your configuration file.
Alternative: Train crYOLO using the command line
Alternative: Train crYOLO using the command line
Navigate to the folder with config_cryolo.json
file, train_image
folder, etc.
Train your network with 5 warmup epochs in GPU 0:
cryolo_train.py -c config.json -w 5 -g 0
The final model file will be written to disk.
cryolo_train.py -c config.json -w 3 -g 0 -e 15
to the training command.