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))
|
||||
|
||||
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:
|
||||
if not params.use_averaged_model:
|
||||
if params.iter > 0:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user