mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
4b38638be8
commit
a51a122846
Binary file not shown.
@ -1604,7 +1604,7 @@ def run_adapter(rank, world_size, args, wb=None):
|
|||||||
)
|
)
|
||||||
'''
|
'''
|
||||||
|
|
||||||
prompt = torch.randn((50, 512), requires_grad=True)
|
prompt = torch.randn((100, 512), requires_grad=True)
|
||||||
optimizer_adapter = ScaledAdam(
|
optimizer_adapter = ScaledAdam(
|
||||||
[prompt],
|
[prompt],
|
||||||
lr=params.adapter_lr,
|
lr=params.adapter_lr,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user