Model Training and Hardware Requirements?

dherath / CFGExplainer

[code] "CFGExplainer: Explaining Graph Neural Network-Based Malware Classification from Control Flow Graphs" by Jerome Dinal Herath, Priti Prabhakar Wakodikar, Ping Yang and Guanhua Yan. In: IEEE/IFIP International Conference on Dependable Systems and Networks (DSN) (2022)

30 stars 5 forks source link

Model Training and Hardware Requirements? #2

Open kevin3567 opened 5 months ago

kevin3567 commented 5 months ago

Hi,

Could I ask:

what GPU is necessary for model training?
what approximately how long does it take to train the model using the GPU mentioned above (and the dataset provided in the repo)?

Thanks in advance.

Edit: Nevermind. Found the answer by referring to the paper (Section V and Table IV). Thanks.

kevin3567 commented 5 months ago

A follow up question:

Does the 2-hour 11-minute CFG training time include the GCN training time? If not, how long does it take to training the GCN model for acquiring node encoding?