Skip to content

Fix beam search output_log_prob index error#732

Open
cpm0722 wants to merge 1 commit into
NVIDIA:mainfrom
cpm0722:main
Open

Fix beam search output_log_prob index error#732
cpm0722 wants to merge 1 commit into
NVIDIA:mainfrom
cpm0722:main

output_log_probs's shape should be (gen_length, batch_size *

00997c9
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs