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
385 stars 81 forks source link

VRAM needed for finetuning #63

Closed miqsoft closed 1 year ago

miqsoft commented 1 year ago

Hey,

Since in the requirements the GPU: Tesla V100S is mentied I'm wondering if this is meant for training the model or for finetuning. Or in other words, is it also possible to do the finetuning on a graphic card with 12GB or so.

Greetings, Michael

zxxcjy123 commented 1 year ago

微调够的

zxxcjy123 commented 1 year ago

微调阶段batch_size32大概需要8-9G显存