Update model.py to fix "IndexError: index 1 is out of bounds for dimension ...

Isuxiz Slidder

2025-03-31 3df109adfccedeb134dea4ba2ea9a2da89872048

 funasr/optimizers/sgd.py

@@ -1,5 +1,4 @@
import torch
from typeguard import check_argument_types


class SGD(torch.optim.SGD):
@@ -21,7 +20,6 @@
        weight_decay: float = 0.0,
        nesterov: bool = False,
    ):
        assert check_argument_types()
        super().__init__(
            params,
            lr=lr,