Skip to content
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

Full Protein -> Logits scoring! Woohoo. #53

Merged
merged 11 commits into from
Nov 26, 2024
Merged

Full Protein -> Logits scoring! Woohoo. #53

merged 11 commits into from
Nov 26, 2024

Conversation

zachcp
Copy link
Owner

@zachcp zachcp commented Nov 26, 2024

cargo test -p ferritin-featurizers test_cli_command_run_example_01 -- --nocapture
Ok(
 ScoreOutput {
    s: Tensor[dims 1, 93; u32], 
    log_probs: Tensor[dims 1, 93, 21; f32], 
    logits: Tensor[dims 1, 93, 21; f32], 
   decoding_order: Tensor[dims 1, 93; u32] })

Mostly the issue is making the broadcasting dimensions explicit.

@zachcp zachcp merged commit 6cc84b5 into main Nov 26, 2024
1 check failed
@zachcp zachcp deleted the 20241126-gather-fix branch November 26, 2024 20:48
@zachcp zachcp mentioned this pull request Nov 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant