UnboundLocalError: cannot access local variable 'current_batch' where it is not associated with a value
wandb:
wandb: You can sync this run to the cloud by running:
wandb: wandb sync logs/wandb/offline-run-20250602_105829-us8eoggp
Shutting down trainer...
Terminated