pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.5k stars 608 forks source link

Fix error when testing xla in cosine similarity metric #3213

Closed kzkadc closed 3 months ago

kzkadc commented 4 months ago

Fixes https://github.com/pytorch/ignite/pull/3203#issuecomment-2009094503

Description:

Check list: