Open DrugD opened 1 week ago
您好。不太清楚这里应该怎么修改? 我没理解错的话: python trainper.py --data-path data/chembl_02 --save-directory train_transformer --model-choice transformer transformer
这是预训练代码,但是在代码模型中需要一个model-pth: def init(self): super(EncoderDecoder, self).init() self.models=self.load_from_file('./experiments/train_transformer/checkpointpretrain/model.pt')
# self.decode = self.models.decode
self.src_embed = self.models.src_embed
self.tgt_embed = self.models.tgt_embed
self.generator = self.models.generator
不太清楚这里的 self.models=self.load_from_file('./experiments/train_transformer/checkpointpretrain/model.pt') 从何而来
哥们,你运行代码了吗?这个根本不影响代码的运行。这个是我留着后续扩展模型的时候使用的。
肯定运行了的,不然怎么定位这里呢?你把github下载下来看看呢
这个代码,至今仍在使用,运行良好,并为合作者提供分子优化服务。
如果你在运行代码的过程有问题,可以提出问题,或者检查环境配置。