RecBole icon indicating copy to clipboard operation
RecBole copied to clipboard

[💡SUG] 关于torch.2.0的 model = torch.compile(model)问题

Open Milkyorange opened this issue 1 year ago • 1 comments

您希望添加的功能是否与某个问题相关? 关于这个问题的简洁清晰的描述,例如,当 [...] 时,我总是很沮丧。 pytorch2.0可以借助 model = torch.compile(model) 来加速训练 请问 recbole如何借助此项功能 描述您希望的解决方案 关于解决方案的简洁清晰的描述。

描述您考虑的替代方案 关于您考虑的,能实现这个功能的其他替代方案的简洁清晰的描述。

其他 您可以添加其他任何的资料、链接或者屏幕截图,以帮助我们理解这个新功能。

Milkyorange avatar Oct 13 '23 05:10 Milkyorange

@Milkyorange 您可以在自己的recbole文件夹下的quick_start.py文件中进行添加,将model = get_model(config["model"])(config, train_data._dataset).to(config["device"])改为model = torch.compile(...)

BishopLiu avatar Oct 24 '23 06:10 BishopLiu