Skip to content

About CTPN predict #31

@Zhangpeixiang

Description

@Zhangpeixiang

您好,最近准备复线一下您的代码,关于CTPNpredict方面,我进行训练之后,得到大概66M的参数模型,然后利用您的预测代码,在我将您的GRU替换为CuDNNLSTM之后,系统提升我显存不够,需要6.2G,我是1060ti,6G,所以显存不够,这个是正常现象么?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions