Use `atexit()` to delay the closing wandb to capture complete log.

[dhruvdcoder]

Currently, wandb.finish() is called in the on_end() method of wandb_allennlp callback. This results in a truncated log on wandb.