search

wzzzd / text_classifier_pytorch

基于Pytorch的文本分类框架，支持TextCNN、Bert、Electra等。
58 stars 10 forks source link

GPU free #7

Open dhl961123 opened 1 year ago

dhl961123 commented 1 year ago

如何计算需要预留的GPU大小

wzzzd commented 1 year ago

不同batch_size大小，占用显存都不同。可以命令行 watch -n 1 nvidia-smi 查看G卡的占用情况，可以根据实际需要调整batch_size大小。