Reference for ultralytics/utils/callbacks/wb.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/callbacks/wb.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.utils.callbacks.wb._log_plots(plots, step)
ultralytics.utils.callbacks.wb.on_pretrain_routine_start(trainer)
Initiate and start project if module is present.
ultralytics.utils.callbacks.wb.on_fit_epoch_end(trainer)
Logs training metrics and model information at the end of an epoch.
Source code in ultralytics/utils/callbacks/wb.py
ultralytics.utils.callbacks.wb.on_train_epoch_end(trainer)
Log metrics and save images at the end of each training epoch.
Source code in ultralytics/utils/callbacks/wb.py
ultralytics.utils.callbacks.wb.on_train_end(trainer)
Save the best model as an artifact at end of training.