Update train_cpu.py - fix heteroscedastic loss clamping

b41eaa4

verified

Jdice27 commited on 15 days ago