GPU's temperature is pretty high, but the GPU memory isn't seem to be occupied (around 200M)

FFrankyy / FINDER

FINDER - FInding key players in complex Networks through DEep Reinforcement learning (Nature Machine Intelligence)

MIT License

176 stars 46 forks source link

Open BW15061999 opened 3 years ago

BW15061999 commented 3 years ago

When I tried another deep learning program, the GPU memory has been occupied up to 90%. Is FINDER really not taking up a lot of GPU memory?

FFrankyy commented 3 years ago

Which graph size are you trying on?

BW15061999 commented 3 years ago

I used command CUDA_VISIBLE_DEVICES=0 python train.py in FINDER_CN_COST with BATCH_SIZE = 16 and others parameters remained untouched

FFrankyy commented 3 years ago

What about graph size? How many graph nodes are you training with?