celldreamer/results/test_metrics.json

{
    "model": "celldreamer",
    "checkpoint": "celldreamer/checkpoints/best.pth",
    "test_samples": 18253,
    "metrics": {
        "avg_total_loss": 0.6892188849982682,
        "avg_recon_loss_mse": 0.6890018098837846,
        "avg_dynamics_loss_kl": 21.70746588540244,
        "std_total_loss": 0.03752287398763396
    }
}