模型占内存特别大

ZhuiyiTechnology / roformer-sim

SimBERT升级版（SimBERTv2）！

Apache License 2.0

439 stars 73 forks source link

Open zhurui-xiaozhuzaizai opened 3 years ago

zhurui-xiaozhuzaizai commented 3 years ago

在16G V100上运行，占用内存接近16个G。是模型本身这么大吗？还是为啥呢

bojone commented 3 years ago

tf自动占满所有显存，不管模型多大。这是tf默认设置，跟keras、bert4keras无关，你可以修改为自适应占用量