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()