-
Notifications
You must be signed in to change notification settings - Fork 3
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
GatherND INVALID_ARGUMENT Requested more than 0 entries #5
Comments
Looking at the error, I would guess that you don't have any binary predicates. To fix this, it might be easier to just add 1 binary predicate to learn if you can. Otherwise, you may also be able to comment out: # This is where the error is
perm_binary = tf.gather_nd(
inputs["binary"], repeat_bidxs, batch_dims=1
) # (B, K, V, V-1, P2)
# ---------------------------
# Compute flattened input
in_tensor = flatten_concat(
# You may then remove perm_binary
[perm_nullary, perm_unary, perm_binary], batch_dims=2
) I say may, because I don't remember whether the downstream elements and parameters make computations for binary predicates. |
The expected input to these layers are: {'nullary_preds': (B, P0), 'unary_preds': (B, N, P1), 'binary_preds': (B, N, N-1, P2)} and probably assume they are non-empty tensors. |
I printed the input tensor, and all my parameter settings are the default settings from the original experiment. However, the program doesn't run when I comment out that code snippet. How can I add a binary predicate for learning? |
THe number of nullary, unary and binary relations are specificied with the |
The text was updated successfully, but these errors were encountered: