transformers
Add new dim to `num_items_in_batch` if necessary
#36967
Merged

Commits
  • Add new dim to `num_items_in_batch` if necessary
    regisss committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    regisss committed 1 year ago
  • Unsqueeze only in the DP case
    regisss committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    IlyasMoutawwakil committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    IlyasMoutawwakil committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    IlyasMoutawwakil committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    SunMarc committed 1 year ago
  • Merge branch 'main' into fix_multi_gpu_num_items
    ydshieh committed 1 year ago
Loading