WFCG icon indicating copy to clipboard operation
WFCG copied to clipboard
verified publisher icon raglandman

模型训练占用显存较大

Open IceStreams opened this issue 3 years ago • 0 comments

你好，我使用16G的RTX3080显卡设备试着跑了一下PU数据集和HongHu数据集，发现显存不够。究其原因，一方面是因为将HSI全影像输入，没有做batch_size处理，占用显存较大。另一方面是因为GAT和PAM处计算自注意力机制会占用较大的显存。请问你使用的是多大显存的RTX3090卡呢？

IceStreams avatar Nov 05 '22 03:11 IceStreams