就算10亿条数据,每个有32特征,每个特征用double存,10^9*2^5*8=10^9*2^8  大概需要238G的内存,而你需要用这么多数据训练一个33维的向量,是不是有点数据量过多了呢。 我建议从两个方面走: 1)如果真的要用这么多数据,那么肯定没法读入内存的,那么就麻烦点,把每次训练的batch size搞大点,每次从硬盘读数据,然后训练 2)训练一个33维的向量真的不需要那么多数据,随机抽出一部分数据来训练就足够了
点赞 1

相关推荐

learYuan:🐕看了都摇头
点赞 评论 收藏
分享
牛客网
牛客企业服务