RecBole
RecBole copied to clipboard
[💡SUG] 关于torch.2.0的 model = torch.compile(model)问题
您希望添加的功能是否与某个问题相关? 关于这个问题的简洁清晰的描述,例如,当 [...] 时,我总是很沮丧。 pytorch2.0可以借助 model = torch.compile(model) 来加速训练 请问 recbole如何借助此项功能 描述您希望的解决方案 关于解决方案的简洁清晰的描述。
描述您考虑的替代方案 关于您考虑的,能实现这个功能的其他替代方案的简洁清晰的描述。
其他 您可以添加其他任何的资料、链接或者屏幕截图,以帮助我们理解这个新功能。
@Milkyorange 您可以在自己的recbole文件夹下的quick_start.py文件中进行添加,将model = get_model(config["model"])(config, train_data._dataset).to(config["device"])
改为model = torch.compile(...)
。