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

[li-aolong]

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.