Open lidanxu opened 4 years ago
作者您好,我下载了您的代码以及 纯tensorflow实现的代码,发现这个模型很吃显存,我的环境是 但是只有把batch size设置为1才能跑,否则就超内存,可以请问你们实验环境是怎么样的,batch_size=1学不起来 但是我个人感觉这个模型没有那么复杂,层数也不多,不知道为何这么吃显存
希望您能给我一些帮助,我没搞明白为什么模型跑起来需要那么大的显存 这样我本地实验根本做不起来
万分感谢!
作者您好,我下载了您的代码以及 纯tensorflow实现的代码,发现这个模型很吃显存,我的环境是
但是只有把batch size设置为1才能跑,否则就超内存,可以请问你们实验环境是怎么样的,batch_size=1学不起来
但是我个人感觉这个模型没有那么复杂,层数也不多,不知道为何这么吃显存