mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 02:24:38 +08:00
* changed config graph structure * removed network udner graphs; added new heirarchy trainer * fixed callback in occupancy graph * lint * changed register callback to setter method
1.4 KiB
1.4 KiB
Logging Features
We provide integration with multiple logging interfaces to log images and statistics during training. All of these loggers are configurable via the config:
# e.g. configs/graphs_default.yaml
logging:
steps_per_log: 10 # how often to log losses/images
max_buffer_size: 20 # defines how many steps to average over (if reporting averages)
writer: # definition of all writers you want to use
TensorboardWriter:
log_dir: "./" # must specify the output path for all writers
LocalWriter:
log_dir: "./"
stats_to_track:
[ITER_LOAD_TIME, ITER_TRAIN_TIME, RAYS_PER_SEC, CURR_TEST_PSNR]
max_log_size: 10
Currently, we provide support for the following loggers:
- Tensorboard:
TensorBoardWriter - Weights and Biases:
WandbWriter - Local writer:
LocalWriter
The LocalWriter simply outputs to the terminal. However, you can specify additional parameters to customize the logging.
- stats_to_track: all the stats that you want to print to the terminal (see list under
EventNameinutils/writer.py). You can add or remove any of the defined enums. - max_log_size: how much content to print onto the screen (e.g. only print 10 lines on the screen at a time). If 0, will print everything without deleting any previous lines.