From d357503988293b6635404b7fc4e279021252daa9 Mon Sep 17 00:00:00 2001 From: vince62s Date: Sat, 11 Nov 2023 15:40:25 +0100 Subject: [PATCH] . --- onmt/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onmt/models/model.py b/onmt/models/model.py index 7e4972280f..e99c09276b 100644 --- a/onmt/models/model.py +++ b/onmt/models/model.py @@ -120,7 +120,7 @@ def load_state_dict( offset = 0 buf_list = [] for buf_name, buf in self.named_buffers(): - buf_list.append(buf_name) + buf_list.append(buf_name) for name, module in self.named_modules(): named_buf_and_param = list(module.named_buffers()) + list( module.named_parameters()