transformers
Add new dim to `num_items_in_batch` if necessary
#36967
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
8
Changes
View On
GitHub
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