- avg_loss_train
- avg_loss_val
- avg_non_role_model_embed_loss_train
- avg_non_role_model_embed_loss_val
- avg_non_role_model_g_cos_loss_train
- avg_non_role_model_g_cos_loss_val
- avg_non_role_model_g_mag_loss_train
- avg_non_role_model_g_mag_loss_val
- avg_pred_std_train
- avg_pred_std_val
- avg_role_model_g_cos_loss_train
- avg_role_model_g_cos_loss_val
- avg_role_model_g_mag_loss_train
- avg_role_model_g_mag_loss_val
- avg_role_model_loss_train
- avg_role_model_loss_val
- avg_role_model_mean_pred_loss_train
- avg_role_model_mean_pred_loss_val
- avg_role_model_std_loss_train
- avg_role_model_std_loss_val
- duration_batch_train
- duration_batch_val
- duration_size_train
- duration_size_val
- duration_train
- duration_val
- n_batch_train
- n_batch_val
- n_size_train
- n_size_val
- train_avg_loss
- train_avg_non_role_model_embed_loss
- train_avg_non_role_model_g_cos_loss
- train_avg_non_role_model_g_mag_loss
- train_avg_pred_std
- train_avg_role_model_g_cos_loss
- train_avg_role_model_g_mag_loss
- train_avg_role_model_loss
- train_avg_role_model_mean_pred_loss
- train_avg_role_model_std_loss
- train_duration
- train_duration_batch
- train_duration_size
- train_n_batch
- train_n_size
- val_avg_loss
- val_avg_non_role_model_embed_loss
- val_avg_non_role_model_g_cos_loss
- val_avg_non_role_model_g_mag_loss
- val_avg_pred_std