4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
nerfstudio/docs/tooling/logging.md
evonneng 91ef54963c
refactoring configs; pulled network out of graph; created trainer heirarchy (#51)
* changed config graph structure

* removed network udner graphs; added new heirarchy trainer

* fixed callback in occupancy graph

* lint

* changed register callback to setter method
2022-06-15 09:06:43 -07:00

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:

  1. Tensorboard: TensorBoardWriter
  2. Weights and Biases: WandbWriter
  3. 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 EventName in utils/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.