[WIP] [tx] Fix loss_fn always defaulting to cross_entropy in SkyRL backend#1053

Merged

tyler-griggs merged 4 commits intomainfrom

tyler/fix-loss-fn-dispatch

Feb 10, 2026

Commits on Feb 8, 2026

[tx] Derive loss_fn from batch instead of hardcoded default

EC2 Default User
and
claude
committed
Store loss_fn as strings in PreparedModelPassBatch

EC2 Default User
and
claude
committed

Commits on Feb 10, 2026

Add loss_fn validation and mixed-batch warning

tyler-griggs
and
claude
committed
style: black formatting fix for skyrl_train.py

tyler-griggs
and
claude
committed