Skip to content

fix: use y.device instead of hardcoded .cuda() in data2vec compute_var - #3710

Open
li-lizhe wants to merge 3 commits into
modelscope:mainfrom
li-lizhe:fix/data2vec-compute-var-device
Open

fix: use y.device instead of hardcoded .cuda() in data2vec compute_var#3710
li-lizhe wants to merge 3 commits into
modelscope:mainfrom
li-lizhe:fix/data2vec-compute-var-device

fix(test): resolve compute_var import and f32_worker spawn pickle

944d3d9
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs