ds

游雁

2024-05-20 1e1500adadf5c7ed3622efa0f48f51b48a78b31e

ds

 funasr/train_utils/trainer_ds.py

@@ -577,7 +577,7 @@
            self.val_loss_avg = (
                self.val_loss_avg * batch_idx + loss_dict["loss"].detach().cpu().item()
            ) / (batch_idx + 1)
            if "acc" in stats:
            if "acc" in loss_dict["stats"]:
                self.val_acc_avg = (
                    self.val_acc_avg * batch_idx + loss_dict["stats"]["acc"].detach().cpu().item()
                ) / (batch_idx + 1)
@@ -740,7 +740,7 @@
                self.val_loss_avg = (
                    self.val_loss_avg * batch_idx + loss_dict["loss"].detach().cpu().item()
                ) / (batch_idx + 1)
                if "acc" in stats:
                if "acc" in loss_dict["stats"]:
                    self.val_acc_avg = (
                        self.val_acc_avg * batch_idx
                        + loss_dict["stats"]["acc"].detach().cpu().item()