どのファイルを確認すれば、コードがクラッシュしたときの原因がわかりますか?

実行中のコードのディレクトリー内にある、wandb/run-<date>_<time>-<run-id>/logs にある debug.logdebug-internal.log を確認してください。