-
Notifications
You must be signed in to change notification settings - Fork 108
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Check failed #1
Comments
i hava found the answer from google-research/bert#146 |
but another question is raise! |
Hey @ningshixian you'll need to remove the pooling layers from trainable weights, e.g.
|
2019-04-10 16:33:34.752272: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1053] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 11481 MB memory) -> physical GPU (device: 0, name: Tesla K40m, pci bus id: 0000:82:00.0, compute capability: 3.5) ▽ segment_ids_1:0 input_ids_1:0 input_mask_1:0 label_ids_1:0 (3613, 2) |
thanks...but it does not work |
hello,
I modified a part of the code,
bert_outputs = self.bert(inputs=bert_inputs, signature="tokens", as_dict=True)['sequence_output']
return (max_seq_length, self.output_size)
but when i run the code, i have the following problem!
"F tensorflow/core/framework/tensor_shape.cc:44] Check failed: NDIMS == dims() (2 vs. 4)Asking for tensor of 2 dimensions from a tensor of 4 dimensions
Aborted (core dumped)"
may I know what is the reason?
The text was updated successfully, but these errors were encountered: