Skip to content

training 设置报错 #87

Description

@xy-derrick
def train_recommender(self):
    if hasattr(self.rec_model, 'bert'):
        if os.environ["CUDA_VISIBLE_DEVICES"] == '-1':
            bert_param = list(self.rec_model.bert.named_parameters())
        else:
            bert_param = list(self.rec_model.module.bert.named_parameters())
        bert_param_name = ['bert.' + n for n, p in bert_param]
    else:
        bert_param = []

在这段代码中 bert_param = list(self.rec_model.module.bert.named_parameters()) 报错说 'TGRecModel' object has no attribute 'module',请问原因是?然后想问一下这个module是干嘛的

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions