search

yunjey / domain-transfer-network

TensorFlow Implementation of Unsupervised Cross-Domain Image Generation
MIT License
861 stars 204 forks source link

MemoryError solver.py load_svhn #9

Open pageedward opened 7 years ago

pageedward commented 7 years ago

File "/home/page/wp/domain-transfer-network/solver.py", line 41, in load_svhn images = np.transpose(svhn['X'], [3, 0, 1, 2]) / 127.5 - 1 MemoryError

Mengmengbai commented 7 years ago

your GPU memory is small

mstfldmr commented 6 years ago

are there any solutions to this? is it possible to solve it with some optimizations?

anmolduainter commented 6 years ago

I also got the same error

anmolduainter commented 6 years ago

is there any solution to this problem?

411066282 commented 6 years ago

images = np.transpose(svhn['X'], [3, 0, 1, 2]) for x in np.nditer(images, op_flags = ['readwrite']):

  `x[...] = x / 127.5 - 1`