Enable wandb to be used for metric logging in pytorch implementation

mattragoza / LiGAN

Deep generative models of 3D grids for structure-based drug discovery

GNU General Public License v2.0

225 stars 44 forks source link

Enable wandb to be used for metric logging in pytorch implementation #24

Closed Dunni3 closed 3 years ago

Dunni3 commented 3 years ago

This PR enables for wandb to be used with the pytorch implementation. It also allows for wandb to be configured via a "wandb" section in the config file.

This implementation only logs metrics to wandb.

mattragoza commented 3 years ago

Looks good to me.