search

aspuru-guzik-group / chemical_vae

Code for 10.1021/acscentsci.7b00572, now running on Keras 2.0 and Tensorflow
Apache License 2.0
470 stars 178 forks source link

"Resource exhausted: OOM when allocating tensor with shape[940,196]" #16

Closed li-aolong closed 5 years ago

li-aolong commented 5 years ago

When I loaded a model in the file intro_to_chemvae.ipynb, it reported as follow:

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[196,196]
     [[Node: z_mean_sample_4/kernel/Assign = Assign[T=DT_FLOAT, _class=["loc:@z_mean_sample_4/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](z_mean_sample_4/kernel, z_mean_sample_4/random_uniform)]]

Is it due to my server's memory？ My server's memory is 16GB, CPU is Intel(R) Xeon(R) CPU E5-1603 v4 @ 2.80GHz.