Michelle0474 cd4534f05a Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

..

encoders

Initial commit

2026-01-18 21:50:21 +08:00

layers

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

__init__.py

Initial commit

2026-01-18 21:50:21 +08:00

benchmark.py

Fallback

2026-02-28 18:29:07 +08:00

final_train_optuna_cv.py

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

heads.py

Initial commit

2026-01-18 21:50:21 +08:00

models.py

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

nested_cv_optuna.py

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

predict.py

加载模型时显式weight_only=False

2026-02-28 16:49:34 +08:00

pretrain.py

统一所有脚本参数统计方式

2026-02-28 17:08:22 +08:00

trainer_balanced.py

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

trainer.py

Add Mixture-of-Experts (MoE) layer support across modeling pipeline

2026-05-26 15:26:18 +08:00

visualization.py

Add loss visualization

2026-01-23 13:40:22 +08:00