Skip to content

Commit

Permalink
Update param_t to ParamsT
Browse files Browse the repository at this point in the history
The type changed
  • Loading branch information
Harrolee authored Jul 20, 2024
1 parent b6ba6cb commit 3bddb4a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions sparse_autoencoder/optimizer/adam_with_reset.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from torch import Tensor
from torch.nn.parameter import Parameter
from torch.optim import Adam
from torch.optim.optimizer import params_t
from torch.optim.optimizer import ParamsT

from sparse_autoencoder.tensor_types import Axis

Expand All @@ -35,7 +35,7 @@ class AdamWithReset(Adam):

def __init__( # (extending existing implementation)
self,
params: params_t,
params: ParamsT,
lr: float | Float[Tensor, Axis.names(Axis.SINGLE_ITEM)] = 1e-3,
betas: tuple[float, float] = (0.9, 0.999),
eps: float = 1e-8,
Expand Down

0 comments on commit 3bddb4a

Please sign in to comment.