train_loss_avg train_acc_avg

游雁

2024-04-30 f5958e96da05220c9e07d0af63df339cae4cc6f7

train_loss_avg train_acc_avg

 funasr/datasets/sense_voice_datasets/datasets.py

@@ -186,7 +186,7 @@
                )

        if self.batch_type != "example":
            for i in range(3):
            for i in range(10):
                outputs = self._filter_badcase(outputs, i=i)

        return outputs

 funasr/train_utils/trainer.py

@@ -308,6 +308,7 @@
                    checkpoint["step_in_epoch"] if "step_in_epoch" in checkpoint else 0
                )
                self.step_in_epoch = 0 if self.step_in_epoch is None else self.step_in_epoch
                print(checkpoint["train_acc_avg"])
                self.train_acc_avg = (
                    checkpoint["train_acc_avg"] if "train_acc_avg" in checkpoint else 0
                )