Run2 run unit test and save results

1. Execute the test script: python /root/SimPO/unit_test/unit_test_1.py.
2. The script will use the simpo_loss implemented in simpo_trainer.py.
3. Capture the generated loss tensor.
4. Save to NPZ:
   • Use numpy.savez to save the results.
   • File path: /root/loss.npz.
   • Key: losses.
5. Ensure the file is correctly written and accessible for the final evaluation.

import numpy as np
import torch

# This logic is typically inside the unit test or triggered by it
# losses = trainer.simpo_loss(input_tensors).detach().cpu().numpy()
# np.savez('/root/loss.npz', losses=losses)