Added support for saving checkpoints in w&b

1 job from master in 3 minutes and 53 seconds (queued for 37 seconds)
Status Job ID Name Coverage
  Test
passed #13051
test

00:03:53