search

Prinsphield / DNA-GAN

DNA-GAN: Learning Disentangled Representations from Multi-Attribute Images
https://openreview.net/pdf?id=rkX1FF_UM
MIT License
65 stars 12 forks source link

core\framework\op_kernel.cc:1192] Resource exhausted: OOM when allocating tensor with shape[64,256,16,16] #2

Open c1a1o1 opened 6 years ago

c1a1o1 commented 6 years ago

core\framework\op_kernel.cc:1192] Resource exhausted: OOM when allocating tensor with shape[64,256,16,16]

When I train it??

Prinsphield commented 6 years ago

Please give more details about your experiment environment. Maybe you can scale down the batch size.

c1a1o1 commented 6 years ago

Caused by op 'decoder_6/deconv2/conv2d_transpose', defined at: File "D:/work/gansgood/DNA-GAN-master/train.py", line 133, in main() File "D:/work/gansgood/DNA-GAN-master/train.py", line 128, in main DNA_GAN = Model(args.attributes, is_train=True) File "D:\work\gansgood\DNA-GAN-master\model.py", line 26, in init self.build_model() File "D:\work\gansgood\DNA-GAN-master\model.py", line 224, in build_model self.Bxs = [self.joiner('decoder', enc_B) for enc_B in self.enc_Bxs] File "D:\work\gansgood\DNA-GAN-master\model.py", line 224, in self.Bxs = [self.joiner('decoder', enc_B) for enc_B in self.enc_Bxs] File "D:\work\gansgood\DNA-GAN-master\model.py", line 174, in joiner strides=[1,2,2,1]) File "D:\work\gansgood\DNA-GAN-master\model.py", line 94, in make_deconv_bn output_shape=out_shape, strides=strides, padding='SAME'

c1a1o1 commented 6 years ago

gtx1070

Prinsphield commented 6 years ago

I have tested it in different linux machines but no windows machine. I sorry for being not able to reproduce that. Maybe you can modify the dataset.py and change the batch size to a smaller number (e.g., 16) and start to train with respect to no more than 2 attributes.