亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

當使用 shape[3075200,512] 分配張量并鍵入 float on 時出現 OOM:

當使用 shape[3075200,512] 分配張量并鍵入 float on 時出現 OOM:

斯蒂芬大帝 2023-10-05 17:04:35
我正在嘗試創建灰度圖像以使用 CNN 模型(使用 keras)。我必須嘗試使用 512*512 的圖像大小,但出現內存溢出錯誤。我無法再減小圖像大小,如果我查看 gpustat,它是 32GB,但我不知道為什么會彈出內存錯誤。我不知道在哪里以及如何找到和增加可分配的 GPU 內存。以下是我迄今為止所做的嘗試。1.減少層中神經元的數量 2.使用較小的批量大?。ó斍芭看笮?) 3.增加要使用的Maxpooling大小 4.增加步幅大小 5.Dropout 6.config = tf.ConfigProto() config.gpu_options .allow_growth = True 7. config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction 8.strategy = tf.distribute.MirroredStrategy() 9.mirrored_strategy = tf.distribute.MirroredStrategy(devices=["/GPU:0", " /XLA_GPU:1"])我已經嘗試了以上所有方法,但它不起作用。如果您知道如何解決,請回答。您無法縮小圖像尺寸。(這就是我實驗的目的)并且它說GPU內存是32GB,但是如果你讓我知道為什么在不是32GB時會出現錯誤,我將非常感激。
查看完整描述

1 回答

?
一只萌萌小番薯

TA貢獻1795條經驗 獲得超7個贊

簡短的回答。停止使用Flatten并將其更改為GlobalAveragePooling2DGlobalMaxPooling2D。

長答案。您嘗試使用的64*64*800完全連接層的輸入大小太大了?,F代CNN模型都不再使用Flatten并取而代之,GlobalAveragePooling2D因為它還保留了CNN的可變尺寸能力。


查看完整回答
反對 回復 2023-10-05
  • 1 回答
  • 0 關注
  • 137 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號