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
8513a0d635
commit
7be2b54d72
Binary file not shown.
@ -697,7 +697,8 @@ def main():
|
|||||||
lora_modules.append(LoRAHook(module))
|
lora_modules.append(LoRAHook(module))
|
||||||
|
|
||||||
for i, lora in enumerate(lora_modules):
|
for i, lora in enumerate(lora_modules):
|
||||||
|
lora_param = torch.load(f"{params.exp_dir}/lora_{i}.pt")
|
||||||
|
lora.lora.load_state_dict(lora_param)
|
||||||
else:
|
else:
|
||||||
if not params.use_averaged_model:
|
if not params.use_averaged_model:
|
||||||
if params.iter > 0:
|
if params.iter > 0:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user