Merge pull request #958 from alibaba-damo-academy/chenmengzheAAA-patch-5

chenmengzheAAA

2023-09-14 61f12d548a074b269312076d3b92a05d5797e57a

 funasr/train/trainer.py

@@ -369,7 +369,7 @@
                            ],
                            "scaler": scaler.state_dict() if scaler is not None else None,
                            "ema_model": model.encoder.ema.model.state_dict()
                            if hasattr(model.encoder, "ema") and model.encoder.ema is not None else None,
                            if hasattr(model, "encoder") and hasattr(model.encoder, "ema") and model.encoder.ema is not None else None,
                        },
                        buffer,
                    )