Two small bugs

[dfulu]

Pull Request

Description

This is to fix two small bugs.

• One is to make sure the predictions and target to the CPU before trying to move to numpy
• The other is to make sure the wandb logger has been instantiated and so has an ID before using this to set the checkpoint path.

[codecov[bot]]

Codecov Report

Attention: 1 lines in your changes are missing coverage. Please review.

Comparison is base (48be53b) 57.65% compared to head (298e10c) 57.62%. Report is 1 commits behind head on main.

Files	Patch %	Lines
pvnet/training.py	0.00%	1 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #114 +/- ## ========================================== - Coverage 57.65% 57.62% -0.04% ========================================== Files 26 26 Lines 1705 1706 +1 ========================================== Hits 983 983 - Misses 722 723 +1 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.