Add W&B Logging - Githubissues

yenchenlin / nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

MIT License

5.37k stars 1.04k forks source link

Add W&B Logging #61

Closed SauravMaheshkar closed 2 years ago

SauravMaheshkar commented 2 years ago

This PR aims to add basic Weights and Biases Metric Logging by appending to the existing training scripts with minimal changes.

The PR adds 3 extra arguments namely --use_wandb, --wandb_project and --wandb_entity which can be used to specify whether to use wandb, the name of the project to be used ("nerf-pytorch" by default) and name of the entity to be used.

SauravMaheshkar commented 2 years ago

Request for Review @yenchenlin

yenchenlin commented 2 years ago

Hello @SauravMaheshkar I don't plan to make wandb a default dependency of this project despite it's an awesome tool for logging!