Set Up W&B and Run Pod During Training