search

linwhitehat / ET-BERT

The repository of ET-BERT, a network traffic classification model on encrypted traffic. The work has been accepted as The Web Conference (WWW) 2022 accepted paper.
MIT License
384 stars 81 forks source link

4090单卡微调提示Cuda Out of memory #89

Closed MiloSciA closed 1 month ago

MiloSciA commented 1 month ago

大佬您好，我看您的论文里面写了微调的时候单卡就可以，想问一下单卡是选用哪个型号呀，我用4090单卡微调爆显存了ORZ

linwhitehat commented 1 month ago

大佬您好，我看您的论文里面写了微调的时候单卡就可以，想问一下单卡是选用哪个型号呀，我用4090单卡微调爆显存了ORZ

你好，我们使用的是单卡 Tesla V100S进行微调，在论文的环境设置中有介绍。