logger abstraction

[rohansaw]

we currently mix our own logger and wandb logging, which makes the code quite ugly in some party. we sould create a custom logger that implements wandb loggger interface so that we can either log locally or with wandb