原创 pytorch 解決out of memory

主要是針對指定了兩片GPU,但默認使用第一塊的問題。 我在用pytorch跑程序的時候,由於代碼比較大,需要用兩塊板子。但無論是在代碼中加入 os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'

原创 GoogLeNet學習筆記

GoogLeNet論文學習筆記AbstractIntroductionRealted WorkMotivation and High Level ConsiderationsArchitectural DetailsGoogLeN