training 设置报错 #87
Description
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是干嘛的