-
-
Notifications
You must be signed in to change notification settings - Fork 120
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
Add CDSA imputation method #303
base: dev
Are you sure you want to change the base?
Conversation
Initially achieved, not yet perfected
Update class names and comments
self.d_k = d_k | ||
self.d_v = d_v | ||
|
||
# w_qs, w_ks for Time Dimension |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
w_qs_time, w_ks_time for the time dimension and w_qs_feature, w_ks_feature for the feature dimension are not used in the func forward().
residual = v | ||
|
||
# now separate the last dimension of q, k, v into different heads -> [batch_size, n_steps, n_heads, d_k or d_v] | ||
q_time = self.w_qs(q_time).view(batch_size, n_steps, self.n_heads, self.d_k) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
self.w_qs and w_ks are not initialized in the func init().
This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity. |
Initially achieved, not yet perfected
What does this PR do?
Add CDSA imputation method.
Initially achieved, not yet perfected.
Before submitting