Bugfix: Correct loss labels when graphing #518

Job	Run time
pip (ubuntu-latest, cpu)	1m 21s
conda (ubuntu-latest, nvidia)	2m 5s
pip (windows-latest, cpu)	7m 38s
conda (ubuntu-latest, cpu)	28s
pip (windows-latest, directml)	7m 34s
conda (macos-latest, nvidia)	1m 28s
conda (macos-latest, cpu)	1m 35s
conda (windows-latest, nvidia)	5m 34s
conda (windows-latest, cpu)	4m 1s
conda (ubuntu-latest, rocm)	1m 0s
conda (windows-latest, directml)	5m 2s
	37m 46s

Provide feedback