Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于数据文件后缀名问题 #20

Open
lpdbz opened this issue Jun 16, 2024 · 1 comment
Open

关于数据文件后缀名问题 #20

lpdbz opened this issue Jun 16, 2024 · 1 comment

Comments

@lpdbz
Copy link

lpdbz commented Jun 16, 2024

我想请问一下:
*.feat1CLS后缀名表示项目文本的bert表示,
*.inter表示什么?
*.index表示原始ID的重映射。
*.pth表示训练模型后缀名文件吗?

不知道我理解的对不对,如果不对请告诉我一下,好嘛?

Snipaste_2024-06-16_13-55-27
@hyp1231
Copy link
Member

hyp1231 commented Jun 16, 2024

你好!

  • .inter 代表用户和商品的交互(interactions),具体可以参考 RecBole 数据格式
  • .pth 是训练后存储的模型参数,具体可参考:

    UniSRec/trainer.py

    Lines 36 to 50 in 05aa5cb

    def save_pretrained_model(self, epoch, saved_model_file):
    r"""Store the model parameters information and training information.
    Args:
    epoch (int): the current epoch id
    saved_model_file (str): file name for saved pretrained model
    """
    state = {
    'config': self.config,
    'epoch': epoch,
    'state_dict': self.model.state_dict(),
    'optimizer': self.optimizer.state_dict(),
    }
    torch.save(state, saved_model_file)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants