File size: 1,013 Bytes
c1a46f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | """
单元测试 — 训练模块 (Person C 实现后需通过)
"""
import pytest
import torch
class TestLabelSmoothedLoss:
"""测试标签平滑损失。"""
def test_no_smoothing_equals_ce(self):
"""smoothing=0 时应等于标准交叉熵。"""
pass
def test_smoothing_reduces_confidence(self):
"""标签平滑应降低损失对正确预测的自信度。"""
pass
def test_ignore_padding(self):
"""应忽略 padding 位置的损失。"""
pass
class TestOptimizer:
"""测试优化器构建。"""
def test_build_adamw(self):
"""测试构建 AdamW。"""
pass
def test_weight_decay_groups(self):
"""测试权重衰减参数组正确。"""
pass
class TestScheduler:
"""测试学习率调度器。"""
def test_warmup_phase(self):
"""测试 warmup 阶段 lr 线性增长。"""
pass
def test_decay_phase(self):
"""测试衰减阶段 lr 下降。"""
pass
|