現在、GANを使おうとしています。そこでデータセットのcsv(50000300ほど)を読み込んだところメモリエラーとなりました。しかし、手違いで本来のcsvを転置したもの(30050000)を読み込んだところ、メモリエラーとならずに生成が開始されました。同じ要領のデータを読み込むことができているのでPCのメモリ自体は足りていると思うのですが、原因として考えられることを教えていただきたいです。
読み込み方法はlist,pandas,numpyを試しましたが全て同様にメモリエラーが出ました。
また、GeneratorをGPUへ転送する際(学習前)にメモリエラーとなっているためバッチサイズは関係ないと思われます。
コードは以下を大元としています。
https://github.com/pytorch/examples/tree/master/dcgan
これにデータセットクラスを加え、上記のようなcsv読み込みをしています。
generatorはソースコードの畳み込み部を全て線形結合にしています。また、入出力は全て列数(50000ほど)としています。
Linear(_len,_len)
ReLU()
Dropout()
……
といった流れです。
あなたの回答
tips
プレビュー