Weights & Biases logger for MAE-pytorch

[morganmcg1]

Hi @pengzhiliang

I am an ML Engineer at Weights & Biases and I wanted to know if you were actively reviewing PRs at the moment? We would love to make a PR to add Weights & Biases' experiment tracking and image logging if you have the time to review it?

I think adding experiment tracking will help you and your users keep the fine-tuning experiments organized and image logging will be helpful in visualising and storing the image reconstructions for better model evaluation :)

We have built integrations into transformers, YOLOv5, PyTorch Lightning etc so we should be able to make a quick and clean PR that shouldn't take too much time for you to review.

[pengzhiliang]

I am sorry for the late response.

Of course you can implement it and pull a request.

I guarantee that I will immediately review it.

Thank you very much!

[jbschiratti]

@morganmcg1 I would also be interested in a W&B logger for this repo.